python - 如何使用 Faster Rcnn 在 tensorflow 对象检测 api 中显示纪元数？

Question

我正在使用 Tensorflow 对象检测 api。我在阅读faster_rcnn_inception_v2_pets.config文件时所理解的是 num_steps 是指总步数而不是历元。但是，那么指定 batch_size 的意义何在？假设我的训练数据中有 500 张图像，并且我设置了批量大小 = 5 和 num_steps = 20k。这是否意味着时代数等于 200 ？

当我运行 model_main.py 时，它只显示 global_steps 损失。因此，如果这些全局步骤不是时期，那么我应该如何更改代码以在每个步骤之后以及每个时期之后显示训练损失和验证损失。

标签： pythontensorflowobject-detection-api

Answer 1

所以你的假设是正确的，你有 200 个时期。

我有一个类似的问题，没有显示损失。我的解决方案是转到 model_main.py 文件，然后插入

tf.logging.set_verbosity(tf.logging.INFO)

进口东西之后。然后它会显示每 100 步后的损失。如果您想在每个时代之后拥有它，您可以更改 set_verbosity ;)