【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
发布了71 篇原创文章 · 获赞 56 · 访问量 9万+

猜你喜欢

转载自blog.csdn.net/baidu_26646129/article/details/90476542