《tf object detection API使用教程--第二讲》利用tensorflow的API,把VOCtrainval_11-May-2012数据集转换成tf.record格式的数据

  • 进入到tensorflow/models/research路径下
  • 执行命令wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar,下载数据集VOC2012
  • 解压数据集 tar -xvf VOCtrainval_11-May-2012.tar
  • python object_detection/dataset_tools/create_pascal_tf_record.py \
        --label_map_path=object_detection/data/pascal_label_map.pbtxt \
        --data_dir=VOCdevkit --year=VOC2012 --set=train \
        --output_path=pascal_train.record
    python object_detection/dataset_tools/create_pascal_tf_record.py \
        --label_map_path=object_detection/data/pascal_label_map.pbtxt \
        --data_dir=VOCdevkit --year=VOC2012 --set=val \
        --output_path=pascal_val.record
  • tensorflow/models/research路径下,就会得到pascal_train.record和pascal_val.record两个文件,至此转换已经成功了。

 

note:由于我的tensorflow版本是1.8,CUDA为9.0,cuDNN为7.0.5,在转换的过程中出现一些警告就是:

/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/utils/dataset_util.py:75: FutureWarning: The behavior of this method will change in future versions. Use specific 'len(elem)' or 'elem is not None' test instead.
  if not xml:
 

1. 这个警告我不知道怎么去掉,要是按照提示去修改,会发生错误。所以警告就警告,不用管他,转换还是会成功的。

2. 我也试着去变换tensorflow到1.9.0版本还是会出现这个警告,所以目前我是不管它了,我也不知道怎么解决。

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值