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

版权声明:版权归原作者所有。未经允许,严禁转载。 https://blog.csdn.net/luccao/article/details/81944174
  • 进入到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版本还是会出现这个警告,所以目前我是不管它了,我也不知道怎么解决。

猜你喜欢

转载自blog.csdn.net/luccao/article/details/81944174