Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "a540c3d78e4ae04cee0bbec2dfce12a740e95eb2"
Commit 0a52f120 authored by Cristina Vasconcelos's avatar Cristina Vasconcelos Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 342136176
parent 31e4a64d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/image_classification/preprocessing.py
View file @ 0a52f120
...@@ -379,12 +379,12 @@ def preprocess_for_train(image_bytes: tf.Tensor, ...@@ -379,12 +379,12 @@ def preprocess_for_train(image_bytes: tf.Tensor,
""" """
images = decode_crop_and_flip(image_bytes=image_bytes) images = decode_crop_and_flip(image_bytes=image_bytes)
images = resize_image(images, height=image_size, width=image_size) images = resize_image(images, height=image_size, width=image_size)
if augmenter is not None:
images = augmenter.distort(images)
if mean_subtract: if mean_subtract:
images = mean_image_subtraction(image_bytes=images, means=MEAN_RGB) images = mean_image_subtraction(image_bytes=images, means=MEAN_RGB)
if standardize: if standardize:
images = standardize_image(image_bytes=images, stddev=STDDEV_RGB) images = standardize_image(image_bytes=images, stddev=STDDEV_RGB)
if augmenter is not None:
images = augmenter.distort(images)
if dtype is not None: if dtype is not None:
images = tf.image.convert_image_dtype(images, dtype) images = tf.image.convert_image_dtype(images, dtype)
......
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