tf2.x 大规模npy文件写入TFRecord
tf.Example 数据类型构建
# array中将.flatten可看成生成了list故value=;
# 而label为int,要生成list故value=[]
def serialize_example(array, label):
feature = {
'array': tf.train.Feature(float_list=tf.train.FloatList(value=raw_image.flatten())),
'label': tf.tr
原创
2020-12-09 14:38:49 ·
290 阅读 ·
0 评论