Tensorflow对象检测训练自己的数据集

Question

我正在尝试为我的AI研究训练一个对象检测模型，

在conda cmd中的代码下运行时，出现以下错误

python train.py --logtostdrr --train_dir=training/ --pipeline=_config_path=training/ssd_mobilenet_pets.config

文件 “ .. \ Anaconda3 \ lib \ site-packages \ object_detection-0.1-py3.6.egg \ object_detection \ data_decoders \ tf_example_decoder.py”， 第29行，在 从object_detection.protos导入input_reader_pb2 ImportError：无法导入名称'input_reader_pb2'

Answer 1

运行protoc命令后将生成pb2文件。在models/research目录中运行以下命令：

protoc object_detection/protos/*.proto --python_out=.

然后，安装研究模块和苗条模块，然后更新您的PYTHONPATH变量：

python setup.py install 
cd slim && python setup.py install 
# Change /path/to/models to your path to models project
export PYTHONPATH=$PYTHONPATH:/path/to/models

完成这些步骤后，您应该能够运行脚本。