4

最近我正在尝试使用 TF2 的 Tensorflow Object Detection API V2，并且我已经成功地训练了一个网络。

但是，我几乎没有找到有关如何保存最佳模型的信息。

我发现了一些简单的教程，它们并没有过多地讨论这些细节。

我在这里发现了一个类似的问题，但是最好的答案是修改legacy/trainer.py，这似乎已经过时了。

我还在这里找到了另一个类似的问题，但是答案正在谈论exporters，它在model_lib.py，但我找不到类似的参数model_lib_v2.py

那么，对于 V2，我应该修改哪个文件来保存最佳模型？

4

1 回答 1

3

假设您使用的是TF2 对象检测存储库中的官方脚本，则有一个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

于 2020-08-26T18:04:35.923 回答