Ошибка API обнаружения объектов: "ImportError: невозможно импортировать имя anchor_generator_pb2"

Question

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

python object_detection/builders/model_builder_test.py

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

from object_detection.protos import anchor_generator_pb2
ImportError: cannot import name anchor_generator_pb2

Я заглянул внутрь object_detection.protos, и, похоже, ничего с именем anchor_generator_pb2 не существует. Кому-нибудь еще удалось заставить эту команду работать или удалось решить эту проблему?

Answer 1

Пропущен шаг в инструкциях по установке, где необходимо выполнить следующее:

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

Answer 2

Это сработало для меня. Запустите его из своегоmodels/research

      python setup.py
protoc -I=./ --python_out=./ ./object_detection/protos/*.proto
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim