Skip to content

GitLab

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,
"""
images = decode_crop_and_flip(image_bytes=image_bytes)
images = resize_image(images, height=image_size, width=image_size)
if augmenter is not None:
images = augmenter.distort(images)
if mean_subtract:
images = mean_image_subtraction(image_bytes=images, means=MEAN_RGB)
if standardize:
images = standardize_image(image_bytes=images, stddev=STDDEV_RGB)
if augmenter is not None:
images = augmenter.distort(images)
if dtype is not None:
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