TF对象检测API:将指标保存到磁盘

时间:2019-12-29 04:16:54

标签: python tensorflow object-detection-api

在对象检测API培训的每个评估步骤之后,我都尝试将评估指标转储到磁盘上。例如,将AP转储到带有评估步骤名称的model_dir中。该框架使用tf.estimator.train_and_evaluate。因此,我最好的猜测是应该使用tf.estimator.EstimatorSpectf.estimator.EvalSpectf.train.SessionRunHook创建一个钩子。

但是,我怀疑这是否是正确的方向。在计算完成时,默认情况下会打印对象检测指标(即AP),并且似乎是在TF图之外计算的。我不能从那里转储它,因为在该调用期间无法访问global_step和model_dir。

0 个答案:

没有答案