bugfix

553dccfa · Yuxin Wu · 16f8f752 · 553dccfa
Commit 553dccfa authored Jul 27, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/ResNet/imagenet_resnet_utils.py examples/ResNet/imagenet_resnet_utils.py +1 -1

No files found.
--- a/examples/ResNet/imagenet_resnet_utils.py
+++ b/examples/ResNet/imagenet_resnet_utils.py
@@ -153,7 +153,7 @@ def eval_on_ILSVRC12(model, model_file, dataflow):

 def image_preprocess(image, bgr=True):
    if image.dtype.base_dtype != tf.float32:
-        image = tf.case(image, tf.float32)
+        image = tf.cast(image, tf.float32)
    image = image * (1.0 / 255)

    mean = [0.485, 0.456, 0.406]    # rgb