Skip to content

GitLab

Commit 549d620f authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 466436335
parent 049cac4c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/serving/image_classification.py
View file @ 549d620f
...@@ -40,7 +40,8 @@ class ClassificationModule(export_base.ExportModule): ...@@ -40,7 +40,8 @@ class ClassificationModule(export_base.ExportModule):
def _build_inputs(self, image): def _build_inputs(self, image):
"""Builds classification model inputs for serving.""" """Builds classification model inputs for serving."""
# Center crops and resizes image. # Center crops and resizes image.
image = preprocess_ops.center_crop_image(image) if self.params.task.train_data.aug_crop:
image = preprocess_ops.center_crop_image(image)
image = tf.image.resize( image = tf.image.resize(
image, self._input_image_size, method=tf.image.ResizeMethod.BILINEAR) image, self._input_image_size, method=tf.image.ResizeMethod.BILINEAR)
......
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