Неполная ошибка формы при попытке экспортировать вывод в api обнаружения объекта тензорного потока, хотя перезаписать параметр input_shape - PullRequest
0 голосов
/ 04 марта 2019

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

python object_detection / export_inference_graph.py --input_type = image_tensor --pipeline_config_path = ... / pipeline.config --trained_checkpoint_prefix = ... / model.ckpt-200000 --output_directory = ... / export-model-200000 --input_shape = 1,300,300,3

в нем говорится "статистика 20 флопов без флопов из-за неполных форм".хотя я перезаписываю input_shape с помощью 1300,300,3.

Как мне решить эту проблему?

Моя конечная цель - преобразовать эту модель в представление IR, в котором у меня возникли проблемы, связанные с выводом формы при преобразовании модели, тогда как у меня нет проблем с преобразованием исходной модели, которую я использовал для обучения обучению, в IR.

Спасибо

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...