4

Recently I am trying Tensorflow Object Detection API V2 with TF2, and I've sucessfully trained a network.

However, I found little information about how to save the best model.

I've found some simple tutorials that don't talk much about these details.

I've found a similar question here, however the best answer is to modify legacy/trainer.py, which seems to be outdated.

I've also found another similar question here, however the answers are talking about exporters, which is in model_lib.py, but I can't find similar parameter in model_lib_v2.py

So, for V2, which file should I modify to save the best model?

user10253771
  • 670
  • 5
  • 19

1 Answers1

2

Assuming you are using the official scripts from TF2 Object Detection repository, there is a tensorflow/models/research/object_detection/exporter_main_v2.py file for saving models.

Usage:

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

Source: https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html#exporting-a-trained-model

Yasserius
  • 95
  • 1
  • 11