利用Tensorflow API将数据集转换为tfRecord格式

本文使用Tensorflow 目标检测API中的文件进行转换。
目标检测数据集 的转换为例,其他可参照源码同理

  • 进入models/research/object_detection/dataset_tools/目录,里面存放各种数据存放方式的转换文件,本文以create_pascal_tf_record.py为例。
  • 运行方式:参数分别表示数据集位置、年份(其中前两个组合为一个路径且不可省略,默认检索路径如下图所示,具体见源码),最后一个参数表示生成文件的位置及名称。目录文件如下图所示(符合该结构即可):
python
create_pascal_tf_record.py
\
--data_dir=D:/Fils/CUR_WORK/datasets/VOCtrainval_11-May-2012/VOCdevkit
\
--year=VOC2012
\
--output_path=D:/Fils/CUR_WORK/tfrecord_datasets/voc2012/val.record


**注:**如出现 Windows fatal exception: access viola错误时,修改文件create_pascal_tf_record.py中的label_map_path参数默认值,原因是路径有误。根据原路径可直接更改为绝对路径即可正常运行。
调整set参数用来指定生成训练、验证还是测试数据(具体看源码,很简单的)

参考
https://blog.csdn.net/dulingwen/article/details/88789254

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值