制作tensorflow数据集
1首先准备好自己图片数据集,我是用cifar10 的其中五个类别,分别是bird,car ,cat,deer,plane。五个类别数据分开放置。例如:然后就是根据数据集生成tfrecord,生成的是protobuf(二进制文件,加速文件传输和处理速度),代码如下<import tensorflow as tfimport osimport randomimport mathimport sy...
原创
2018-04-13 18:27:36 ·
818 阅读 ·
0 评论