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

Question

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

Обучение использованию детектора пользовательских объектов - руководство по API обнаружения объектов Tensorflow

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

python3 train.py --logtostderr --train_dir=training/ 
                 --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

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

Traceback (most recent call last):
  File "train.py", line 49, in <module>
    from object_detection.builders import dataset_builder
  File "C:\Program Files\Python36\lib\site-packages\slim-0.1-py3.6.egg\object_detection\builders\dataset_builder.py", line 27, in <module>
    from object_detection.data_decoders import tf_example_decoder
  File "C:\Program Files\Python36\lib\site-packages\slim-0.1-py3.6.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

Ваша установка api обнаружения объекта tenorflow не завершена. Вы пропустили этап компиляции protobuf , см. Здесь . Если вы загляните в папку protos специально под object_detection, файла с именем 'input_reader_pb2' нет, но если вы выполнили этап компиляции protobuf, файл будет сгенерирован автоматически! Если ваша установка прошла успешно, она должна пройти тестовую программу test .