Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "a2b07bfde45b9fa473c768ccc4c1d72e84e8a2f0"
Commit 9dc368bc authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 457843445
parent f962ce38
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
official/vision/dataloaders/classification_input.py
View file @ 9dc368bc
...@@ -254,6 +254,11 @@ class Parser(parser.Parser): ...@@ -254,6 +254,11 @@ class Parser(parser.Parser):
return image 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 @classmethod
def inference_fn(cls, def inference_fn(cls,
image: tf.Tensor, image: tf.Tensor,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment