转换tfrecord代码.zip
文件结构如下 ---pic 我们准备的图片 ------train 训练用 ------validation 验证训练结果用 ---data_convert.py 将我们准备的图片,转换为tfrecord ---src 资源工具代码 ------tfrecord.py tfrecord的转换类 其中,data_convert.py是我们转换的主要代码。 使用方法: 在文件夹下,运行: python data_convert.py -t pic/ \ --train-shards 2 \ --validation-shards 2 \ --num-threads
下载地址
用户评论