Как сохранить лучшие модели в последней версии API обнаружения объектов Tensorflow V2 с TF2?

Недавно я пробую Tensorflow Object Detection API V2 с TF2, и я успешно обучил сеть.

Однако я нашел мало информации о том, как сохранить лучшую модель.

Я нашел несколько простых руководств, в которых не много говорится об этих деталях.

Я нашел подобный вопрос здесь, однако лучший ответ изменитьlegacy/trainer.py, который кажется устаревшим.

Я также нашел другой подобный вопрос здесь, однако ответы говорят оexporters, который в model_lib.py, но я не могу найти аналогичный параметр в model_lib_v2.py

Итак, для версии 2, какой файл мне следует изменить, чтобы сохранить лучшую модель?

Источник

1 ответ

Предполагая, что вы используете официальные скрипты из репозитория TF2 Object Detection, существуетtensorflow/models/research/object_detection/exporter_main_v2.py файл для сохранения моделей.

Использование:

python .\exporter_main_v2.py \
    --input_type image_tensor \
    --pipeline_config_path .\...\pipeline.config \
    --trained_checkpoint_dir .\...\training \
    --output_directory .\...\my_model

Источник: https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html

Источник
Другие вопросы по тегам