Skip to content

GitLab

Commit 2cc4b2e2 authored by Marianne Linhares Monteiro's avatar Marianne Linhares Monteiro Committed by GitHub
Browse files

Adding image = tf.cast(image, tf.float32)

parent 751bfb3f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tutorials/image/cifar10_estimator/cifar10.py
View file @ 2cc4b2e2
...@@ -70,6 +70,7 @@ class Cifar10DataSet(object): ...@@ -70,6 +70,7 @@ class Cifar10DataSet(object):
label = tf.cast(features['label'], tf.int32) label = tf.cast(features['label'], tf.int32)
# Custom preprocessing . # Custom preprocessing .
image = tf.cast(image, tf.float32)
image = self.preprocess(image) image = self.preprocess(image)
return image, label return image, label
......
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