问题
Image
和Label
数据成对写入TFRecord
文件,按理训练过程中读取的Image
和Label
也应该是一一对应的,但有的时候发现Image
和Label
并不能匹配。如:
将以下数据写入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
时,获取Image
和Label
的节点计算了两次,而每一次新的读取数据节点计算过程都会读取下一组数据,也就是说两次读取数据的计算过程让Image
和Label
发生了错位,使得一张图片与下一张图片的Label
对应了起来。
image = sess.run([image_features])
label = sess.run([label_features])
解决
想要让Image
和Label
一一对应,只要使用同一个计算节点得到Image
和Label
数据,换言之,只run
一次:
image, label = sess.run([image_features, label_features])
此时你会发现,新得到的图片和Label一一对应了:
Image 1 —— Label 1
Image 2 —— Label 2
Image 3 —— Label 3