Internal change

PiperOrigin-RevId: 457843445

Internal change
PiperOrigin-RevId: 457843445
a3b666b4 · Fan Yang · A. Unique TensorFlower · f101b11a · a3b666b4
Commit a3b666b4 authored Jun 28, 2022 by Fan Yang Committed by A. Unique TensorFlower Jun 28, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

official/vision/dataloaders/classification_input.py official/vision/dataloaders/classification_input.py +5 -0

No files found.
--- a/official/vision/dataloaders/classification_input.py
+++ b/official/vision/dataloaders/classification_input.py
@@ -254,6 +254,11 @@ class Parser(parser.Parser):

    return image

+  def parse_train_image(self, decoded_tensors: Dict[str,
+                                                    tf.Tensor]) -> tf.Tensor:
+    """Public interface for parsing image data for training."""
+    return self._parse_train_image(decoded_tensors)
+
  @classmethod
  def inference_fn(cls,
                   image: tf.Tensor,