ndex1 for i in range(img_list.__len__())]
index1 += 1
img_data += img_list
labels_data += label_list
recorder_file = 'images/train.tfrecord'
writer = tf.python_io.TFRecordWriter(recorder_file)
for i in range(img_data.__len__()):
im_d=img_data[i]
im_l=labels_data[i]
data= cv2.imread(im_d)
#data = tf.gfile.FastGFile(im_d,'rb').read()
#获得的是字节型不用转换,所获得的的包比OpenCV小
#将图片和标签封装
ex = tf.train.Example(
features = tf.train.Features(
feature = {
'image' : tf.train.Feature(
bytes_list=tf.train.BytesList(
value=[data.tobytes()]
)),
'label' : tf.train.Feature(
int64_list=tf.train.Int64List(
value=[im_l]
)),
}
)
)
writer.write(ex.SerializeToString())
print('everything is ok!')
writer.close()
ile "D:/2D/imgWrite.py", line 46, in
value=[data.tobytes()]
AttributeError: 'NoneType' object has no attribute 'tobytes'