1. 首页
  2. 人工智能
  3. 深度学习
  4. 转换tfrecord代码.zip

转换tfrecord代码.zip

上传者: 2020-07-17 04:16:41上传 ZIP文件 7.28KB 热度 19次
文件结构如下 ---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
下载地址
用户评论