如何使用Faster Rcnn在Tensorflow对象检测api中显示时期数?

时间:2019-06-20 13:04:42

标签: python tensorflow object-detection-api

我正在使用Tensorflow对象检测API。我理解阅读 faster_rcnn_inception_v2_pets.config 文件的理解是,num_steps是步骤的总数而不是时期。但是,指定batch_size有什么意义呢?可以说我的训练数据中有500张图像,并且我将批量大小设置为5,num_steps = 20k。那是否意味着时期数等于200 ??

当我运行model_main.py时,它仅显示global_steps损失。因此,如果这些全局步骤不是新纪元,那么我应该如何更改代码以在每个步骤之后以及每个新纪元后显示火车损耗和val损耗。

1 个答案:

答案 0 :(得分:0)

所以您的假设是正确的,即您有200个时期。

我也有类似的问题,没有表现出损失。 我的解决方法是转到model_main.py文件,然后插入

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

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