生成TFRecord文件

时间:2017-08-17 05:18:26

标签: tensorflow object-detection

成功安装了tensorflow和对象检测模块后,我无法生成PASCAL VOC TFRecord文件,详见文档。

从/ models目录

运行
python2 object_detection/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

ImportError:没有名为object_detection.utils的模块

我按照建议将object_detection添加到系统路径中,如下面提到的链接所建议的那样。这似乎暂时解决了这个问题,但在protoc编译的另一个文件中引发了相同的导入错误。我的猜测是导入"来自object_detection.XXX导入YYY"已跨文件使用,并且不可能对所有此类源文件进行更改。

tensorflow object detection API ImportError when generating PASCAL TFRecord files

期待一项决议。提前谢谢。

1 个答案:

答案 0 :(得分:0)

安装和包路径信息清楚地描述为install OD tensorflow

# From tensorflow/models/
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

# Also you can add the same to your ~/bashrc file as 
export PYTHONPATH=$PYTHONPATH:/absolute_path/to/tensorflow/models:/absolute/path/to/tensowflow/models/slim