Skip to content

GitLab

"git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "93024ace026e6e0a30449a932fa30cfd49258251"
Commit 974ae9c4 authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 466436335
parent 90dedf26
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/serving/image_classification.py
View file @ 974ae9c4
...@@ -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