转换代码

python data_convert2tfrecord.py -t pic/ \

 --train-shards 2 \

 --validation-shards 2 \

 --num-threads 2 \

 --dataset-name ……

转换后的结果

1、输出信息结果

INFO:root:2018-09-25 19:31:12.948984: Finished writing all 4800 images in data set.

2、得到的文件：得到转换后的格式文件