#tensorflow #deep-learning #tensorflow2.0
#tensorflow #глубокое обучение #tensorflow2.0
Вопрос:
Я пытаюсь обучить модель обнаружения объектов Tensorflow. Во время обучения, когда я выполняю заключительные команды:
Для tensorflow версии 1.14 : python model_main.py --logtostderr --train_dir="/path to dir training/" --pipeline_config_path="Path to ssd_mobilenet_v1_pets_Colab.config"
Для tensorflow версии 2.0 : python model_main_tf2.py --logtostderr --train_dir="/path to dir training/" --pipeline_config_path="Path to ssd_mobilenet_v1_pets_Colab.config"
В первом случае (tensorflow 1.14) я получаю следующее сообщение об ошибке: ValueError: ssd_mobilenet_v1 is not supported. See model_builder.py for features extractors compatible with different versions of Tensorflow
и для второго (tensorflow 2.0) случая я получаю ValueError: ssd_mobilenet_v1 is not supported. See model_builder.py for features extractors compatible with different versions of Tensorflow
Я перепробовал все возможные подходы, но не понял, что мне делать. PS Я новичок в глубоком обучении.
Комментарии:
1. Где вы загрузили свою модель и можете ли вы поделиться своим файлом конфигурации? Я думаю, вам следует изменить значение функции извлечения в вашем конфигурационном файле, взяв пример из модельного зоопарка API обнаружения объектов: github.com/tensorflow/models/blob/master/research /…
Ответ №1:
Я думаю, что вы используете модели обнаружения объектов Tensorflow 1.Чтобы использовать их, запустите tensorflow версии 1.15.0, он должен работать.Эти модели не будут работать с tensorflow 2, для использования моделей tensorflow 2 используйте tensorflow версии 2.3.0.