Ошибка импорта: невозможно импортировать имя input_reader_pb2

Question

Я использую Tensorflow Object Detection API для обучения моей модели обнаружения объектов.Я собрал набор данных и прохожу учебное пособие this .

Все шло хорошо, пока я не попытался обучить свой набор данных.Когда я запускаю следующую строку в терминале,

python train.py --logtostderr \ 
   --train_dir=training/ \       
--pipeline_config_path=training/ssd_mobilenet_v1_coco.config 

я получаю следующую ошибку

Traceback (most recent call last):
File "legacy/train.py", line 49, in <module>
from object_detection.builders import dataset_builder
File "/usr/local/lib/python2.7/dist-packages/object_detection-0.1- 
py2.7.egg/object_detection/builders/dataset_builder.py", line 27, in 
<module>
from object_detection.data_decoders import tf_example_decoder
File "/usr/local/lib/python2.7/dist-packages/object_detection-0.1- 
py2.7.egg/object_detection/data_decoders/tf_example_decoder.py", line 
24, in <module>
from object_detection.protos import input_reader_pb2
ImportError: cannot import name input_reader_pb2

Я пытался искать по сайтам и комментариям, но, к сожалению, не мог обойтись, может кто-нибудьПожалуйста, помогите!

Спасибо!

Answer 1

Я разобрался с ответом.Я использовал protoc версии 3.0.0 (компилятор protobuf), который находится в режиме бета-тестирования и имеет некоторые ошибки.Итак, сначала проверьте версию protobuf, используя

proto --version

, если это libprotoc 3.0.0, очистите его и, как только сделаете это, установите 2.6.1,

cd models/research
protoc object_detection/ --python.out=.

теперь это должно работать!