BUG- 使用TFRecord 加载数据,程序运行tf.train.shuffle_batch() 报错

这里就不详细讨论TFRecord的作用,以及如何使用,具体的使用方法,请参考文献【1】。

Error:

OutOfRangeError (see above for traceback): RandomShuffleQueue '_4_shuffle_batch/random_shuffle_queue' is closed and has insufficient elements (requested 5, current size 0)
     [[Node: shuffle_batch = QueueDequeueManyV2[component_types=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"](shuffle_batch/random_shuffle_queue, shuffle_batch/n)]]
 

原因:

根据【1】,在数据转成TFRecord之前,数据类型是np.uint8格式;在加载TFRecord格式的数据,并使用tf.decode_raw从中提取数据的时候,数据类型设置为tf.float32;数据类型前后不一致导致的。

参考文献:

【1】https://zhuanlan.zhihu.com/p/53917588

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值