- 进入到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版本还是会出现这个警告,所以目前我是不管它了,我也不知道怎么解决。