Internal change

PiperOrigin-RevId: 452345162

Internal change
PiperOrigin-RevId: 452345162
16ada64d · Fan Yang · A. Unique TensorFlower · bec89e3a · 16ada64d
Commit 16ada64d authored Jun 01, 2022 by Fan Yang Committed by A. Unique TensorFlower Jun 01, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

official/vision/dataloaders/tfexample_utils.py official/vision/dataloaders/tfexample_utils.py +3 -0

No files found.
--- a/official/vision/dataloaders/tfexample_utils.py
+++ b/official/vision/dataloaders/tfexample_utils.py
@@ -159,11 +159,14 @@ def create_distillation_example(
      np.uint8(np.random.rand(image_height, image_width, 3) * 255),
      fmt=image_format)
  soft_labels = [0.6] * num_labels
+  labels = [0]
  serialized_example = tf.train.Example(
      features=tf.train.Features(
          feature={
              IMAGE_KEY: (tf.train.Feature(
                  bytes_list=tf.train.BytesList(value=[image]))),
+              CLASSIFICATION_LABEL_KEY: (tf.train.Feature(
+                  int64_list=tf.train.Int64List(value=labels))),
              DISTILATION_LABEL_KEY: (tf.train.Feature(
                  float_list=tf.train.FloatList(value=soft_labels))),
          })).SerializeToString()