转换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
文件列表
转换tfrecord代码.zip
(预估有个2文件)
转换tfrecord代码
pic
train
validation
data_convert.py
1KB
src
tfrecord.py
18KB
暂无评论