【发布时间】:2015-12-02 18:15:36
【问题描述】:
我通过从 tfrecords 获取想法 here 来读取一批图像(由 this 转换)
我的图像是 cifar 图像 [32, 32, 3],正如您在阅读和传递图像时看到的那样,形状是正常的 (batch_size=100)
据我所知,日志中提到的两个最值得注意的问题是
- 12228 的形状,我不知道我从哪里得到的。我所有的张量的形状都是 [32, 32, 3] 或 [None, 3072]
- 样本用完
Compute status: Out of range: RandomSuffleQueue '_2_input/shuffle_batch/random_shuffle_queue' is closed and has insufficient elements (requested 100, current size 0)
我该如何解决这个问题?
日志:
1- image shape is TensorShape([Dimension(3072)])
1.1- images batch shape is TensorShape([Dimension(100), Dimension(3072)])
2- images shape is TensorShape([Dimension(100), Dimension(3072)])
W tensorflow/core/kernels/queue_ops.cc:79] Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
W tensorflow/core/common_runtime/executor.cc:1027] 0x7fa72abc89a0 Compute status: Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
[[Node: input/shuffle_batch/random_shuffle_queue_enqueue = QueueEnqueue[Tcomponents=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/cpu:0"](input/shuffle_batch/random_shuffle_queue, input/sub, input/Cast_1)]]
W tensorflow/core/kernels/queue_ops.cc:79] Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
W tensorflow/core/common_runtime/executor.cc:1027] 0x7fa72ab9d080 Compute status: Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
[[Node: input/shuffle_batch/random_shuffle_queue_enqueue = QueueEnqueue[Tcomponents=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/cpu:0"](input/shuffle_batch/random_shuffle_queue, input/sub, input/Cast_1)]]
W tensorflow/core/kernels/queue_ops.cc:79] Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
W tensorflow/core/common_runtime/executor.cc:1027] 0x7fa7285e55a0 Compute status: Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
[[Node: input/shuffle_batch/random_shuffle_queue_enqueue = QueueEnqueue[Tcomponents=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/cpu:0"](input/shuffle_batch/random_shuffle_queue, input/sub, input/Cast_1)]]
W tensorflow/core/kernels/queue_ops.cc:79] Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
W tensorflow/core/common_runtime/executor.cc:1027] 0x7fa72aadb080 Compute status: Invalid argument: Shape mismatch in tuple component 0. Expected [3072], got [12288]
[[Node: input/shuffle_batch/random_shuffle_queue_enqueue = QueueEnqueue[Tcomponents=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/cpu:0"](input/shuffle_batch/random_shuffle_queue, input/sub, input/Cast_1)]]
W tensorflow/core/common_runtime/executor.cc:1027] 0x7fa72ad499a0 Compute status: Out of range: RandomSuffleQueue '_2_input/shuffle_batch/random_shuffle_queue' is closed and has insufficient elements (requested 100, current size 0)
[[Node: input/shuffle_batch = QueueDequeueMany[component_types=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/cpu:0"](input/shuffle_batch/random_shuffle_queue, input/shuffle_batch/n)]]
Traceback (most recent call last):
File "/Users/HANEL/Documents/my_cifar_train.py", line 110, in <module>
tf.app.run()
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/platform/default/_app.py", line 11, in run
sys.exit(main(sys.argv))
File "/Users/HANEL/my_cifar_train.py", line 107, in main
train()
File "/Users/HANEL/my_cifar_train.py", line 76, in train
_, loss_value = sess.run([train_op, loss])
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 345, in run
results = self._do_run(target_list, unique_fetch_targets, feed_dict_string)
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 419, in _do_run
e.code)
tensorflow.python.framework.errors.OutOfRangeError: RandomSuffleQueue '_2_input/shuffle_batch/random_shuffle_queue' is closed and has insufficient elements (requested 100, current size 0)
[[Node: input/shuffle_batch = QueueDequeueMany[component_types=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/cpu:0"](input/shuffle_batch/random_shuffle_queue, input/shuffle_batch/n)]]
Caused by op u'input/shuffle_batch', defined at:
File "/Users/HANEL/my_cifar_train.py", line 110, in <module>
tf.app.run()
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/platform/default/_app.py", line 11, in run
sys.exit(main(sys.argv))
File "/Users/HANEL/my_cifar_train.py", line 107, in main
train()
File "/Users/HANEL/my_cifar_train.py", line 39, in train
images, labels = my_input.inputs()
File "/Users/HANEL/my_input.py", line 157, in inputs
min_after_dequeue=200)
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/training/input.py", line 453, in shuffle_batch
return queue.dequeue_many(batch_size, name=name)
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/ops/data_flow_ops.py", line 245, in dequeue_many
self._queue_ref, n, self._dtypes, name=name)
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/ops/gen_data_flow_ops.py", line 319, in _queue_dequeue_many
timeout_ms=timeout_ms, name=name)
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/ops/op_def_library.py", line 633, in apply_op
op_def=op_def)
File "/Users
/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1710, in create_op
original_op=self._default_original_op, op_def=op_def)
File "/Users/HANEL/tensorflow/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 988, in __init__
self._traceback =
_extract_stack()
【问题讨论】:
-
嗨@mrry 是的,我会把它发给你,但我发现了第二个问题,我使用 training_iterations 到 20 小于 100 (batch_size) 导致元素不足。我猜第一个问题是我机器的线程大小,它是 4 线程和 12228 = 4 * 3072
-
最可能的问题是传递给
set_shape()的大小与decode_raw生成的张量的真实大小不匹配 - 也许管道早期出现了问题。要找出真实的形状,您可以执行以下操作:image_shape = tf.shape(image); ...; sess.run(image_shape)以获得真实的形状。 -
查看更多输入代码,看起来您在将图像写入 TFRecord 文件之前将其转换为
np.int32数组:images_only = [np.asarray(image[1],**np.int32**) for image in images]。但是,您将它们读入为tf.uint8值,这意味着您将拥有四倍的值,以及4 * 3072 = 12288。 -
@mrry 非常感谢,它有效。
-
@mrry 让您的评论成为更多积分的答案。你沉迷于积分不是吗?请记住,有些 StackOverflow 用户不阅读 cmets。此外,我不记得 Google 搜索仅基于 cmets 的答案返回结果。
标签: python tensorflow