Merge pull request #3340 from tensorflow/convert_image_dtype-explanation

Add a note about tf.image.convert_image_dtype

Merge pull request #3340 from tensorflow/convert_image_dtype-explanation
Add a note about tf.image.convert_image_dtype
f568ca59 · Neal Wu · GitHub · 40d29d42 · 6eeec4e7 · f568ca59
Unverified Commit f568ca59 authored Feb 07, 2018 by Neal Wu Committed by GitHub Feb 07, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/resnet/imagenet_main.py official/resnet/imagenet_main.py +2 -0

No files found.
--- a/official/resnet/imagenet_main.py
+++ b/official/resnet/imagenet_main.py
@@ -82,6 +82,8 @@ def parse_record(raw_record, is_training):
  image = tf.image.decode_image(
      tf.reshape(parsed['image/encoded'], shape=[]),
      _NUM_CHANNELS)
+
+  # Note that tf.image.convert_image_dtype scales the image data to [0, 1).
  image = tf.image.convert_image_dtype(image, dtype=tf.float32)

  image = vgg_preprocessing.preprocess_image(