【Tensorflow】读取TFRecord文件时,Image和Label无法一一对应

问题

ImageLabel数据成对写入TFRecord文件,按理训练过程中读取的ImageLabel也应该是一一对应的,但有的时候发现ImageLabel并不能匹配。如:

将以下数据写入TFrecord中:

Image 1 —— Label 1
Image 2 —— Label 2
Image 3 —— Label 3

读取TFRecord时,数据发生错位:

Image 1 —— Label 2
Image 2 —— Label 3
Image 3 —— Label 1

原因

以上不匹配情形产生的原因之一是在读取TFRecord时,获取ImageLabel的节点计算了两次,而每一次新的读取数据节点计算过程都会读取下一组数据,也就是说两次读取数据的计算过程让ImageLabel发生了错位,使得一张图片与下一张图片的Label对应了起来。

image = sess.run([image_features])
label = sess.run([label_features])

解决

想要让ImageLabel一一对应,只要使用同一个计算节点得到ImageLabel数据,换言之,只run一次:

image, label = sess.run([image_features, label_features])

此时你会发现,新得到的图片和Label一一对应了:

Image 1 —— Label 1
Image 2 —— Label 2
Image 3 —— Label 3
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值