Tensorflow Object Detection API中`train.py`和`model_main.py`的区别

Question

我通常只是train.py用来训练使用 Tensorflow Object Detection API。但是，我从https://www.kaggle.com/c/rsna-pneumonia-detection-challenge/discussion/68581中读到，您也可以使用它model_main.py来训练模型并在 Tensorboard 上查看实时绘图和图像。

1. 你是如何model_main.py在 Tensorboard 上使用的？
2. train.py和 和有model_main.py什么区别？

Answer 1

1. 在 TensorBoard 上，model_main.py 输出与 train.py 类似的图，但在 model_main.py 中，也测量了模型在评估数据集上的性能。

2. model_main.py 是 TensorFlow 对象检测 API 中的较新版本。它用于训练和评估模型。使用 train.py 时，我们必须运行一个单独的程序进行评估（eval.py），而 model_main.py 执行这两个程序。例如，训练代码将运行一定时间（例如 5 分钟或每 2000 步），然后训练将停止并运行评估。评估结束后，将再次继续培训。然后再次重复相同的循环。