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

2020-11-12  本文已影响0人  是我真的是我

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

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

上一篇 下一篇

猜你喜欢

热点阅读