tensorflow - 如何使用 TF2 在最新的 Tensorflow Object Detection API V2 中保存最佳模型？

Question

最近我正在尝试使用 TF2 的 Tensorflow Object Detection API V2，并且我已经成功地训练了一个网络。
但是，我几乎没有找到有关如何保存最佳模型的信息。
我找到了一些简单的教程，这些教程并没有过多讨论这些细节。
我发现了一个类似的问题 here ，但是最好的答案是修改 legacy/trainer.py ，这似乎已经过时了。
我还发现了另一个类似的问题 here ，然而答案是关于 exporters ，在 model_lib.py ，但我在 model_lib_v2.py 中找不到类似的参数
那么，对于V2，我应该修改哪个文件来保存最好的模型？

Answer 1

假设您使用的是来自 TF2 Object Detection repository 的官方脚本，有一个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#exporting-a-trained-model