Tensorflow对象检测API不显示全局步骤

Question

我是新来的。我最近开始使用对象检测，并决定使用Tensorflow对象检测API。但是，当我开始训练模型时，尽管它仍在后台进行训练，但它并未像应有的那样显示全局步骤。

详细信息： 我正在服务器上训练，并在Windows上使用OpenSSH访问它。我通过收集图片并标记它们来训练自定义数据集。我使用model_main.py对其进行了培训。而且，直到几个月前，API还是有所不同，直到最近它们才更改为最新版本。例如，之前它曾经使用train.py进行训练，而不是model_main.py。我可以找到的所有在线教程都使用train.py，因此最新提交可能存在问题。但是我找不到其他人来解决这个问题。

谢谢！

Answer 1

在tf.logging.set_verbosity(tf.logging.INFO)脚本的import部分之后添加model_main.py。每第100步将显示摘要。

Answer 2

按照Thommy257的建议，在model_main.py的import部分之后添加tf.logging.set_verbosity(tf.logging.INFO)会在默认情况下每100步打印一次摘要。

此外，要指定摘要的频率，请更改

  

config = tf.estimator.RunConfig（model_dir = FLAGS.model_dir）

到

  

config = tf.estimator.RunConfig（model_dir = FLAGS.model_dir，log_step_count_steps = k ）

每k步打印一次。

Answer 3

关于对model_main的最新更改，可以在“旧版”文件夹中找到以前的版本。我使用此旧文件夹中的train.py和eval.py具有与以前相同的功能。