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

Tensorflow object detection API not displaying global steps

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

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

提前致谢!

model_main.py 脚本的导入部分之后添加 tf.logging.set_verbosity(tf.logging.INFO)。它会在每 100 步后显示一个摘要。

按照 Thommy257 的建议,在 model_main.py 的导入部分后添加 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 步后打印的位置。

关于最近对 model_main 的更改,以前的版本可在文件夹 "legacy" 中找到。我使用此旧文件夹中的 train.py 和 eval.py,功能与以前相同。