Commit dee751fa authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #895 from tensorflow/cifar10-inputs

Added shape to cifar10_input.py
parents 2cd62f6c 19c23e72
...@@ -242,6 +242,10 @@ def inputs(eval_data, data_dir, batch_size): ...@@ -242,6 +242,10 @@ def inputs(eval_data, data_dir, batch_size):
# Subtract off the mean and divide by the variance of the pixels. # Subtract off the mean and divide by the variance of the pixels.
float_image = tf.image.per_image_standardization(resized_image) float_image = tf.image.per_image_standardization(resized_image)
# Set the shapes of tensors.
float_image.set_shape([height, width, 3])
read_input.label.set_shape([1])
# Ensure that the random shuffling has good mixing properties. # Ensure that the random shuffling has good mixing properties.
min_fraction_of_examples_in_queue = 0.4 min_fraction_of_examples_in_queue = 0.4
min_queue_examples = int(num_examples_per_epoch * min_queue_examples = int(num_examples_per_epoch *
......
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