Когда я запускаю export_inference_graph.py
с правильными путями из папки object_detection
или из папки models/research
вот так:
python export_inference_graph \ # object_detection/ (if from /research)
--input_type image_tensor \
--pipeline_config_path path/to/ssd_inception_v2.config \
--trained_checkpoint_prefix path/to/model.ckpt \
--output_directory path/to/exported_model_directory
то я получил следующую ошибку:
File "object_detection/export_inference_graph.py", line 71, in <module>
from object_detection import exporter
File "/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/exporter.py", line 28, in <module>
from object_detection.builders import model_builder
File "/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/builders/model_builder.py", line 17, in <module>
from object_detection.builders import anchor_generator_builder
File "/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/builders/anchor_generator_builder.py", line 20, in <module>
from object_detection.protos import anchor_generator_pb2
ImportError: cannot import name anchor_generator_pb2
Внутри anchor_generator_builder.py
он пытается импортировать anchor_generator_pb2
, но файла с таким именем не существует.
Дополнительная информация: - Да, я добавил библиотеки с помощью:
# From tensorflow/models/research/
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
- Да, я вытащил последнюю версию tensorflow/models
Пожалуйста, скажите мне, как правильно экспортировать новый замороженный график, я думаю, это должно быть очень просто, но это останавливает всю мою работу.
Спасибо, ребята, заранее