在对象检测API培训的每个评估步骤之后,我都尝试将评估指标转储到磁盘上。例如,将AP转储到带有评估步骤名称的model_dir中。该框架使用tf.estimator.train_and_evaluate
。因此,我最好的猜测是应该使用tf.estimator.EstimatorSpec
为tf.estimator.EvalSpec
或tf.train.SessionRunHook
创建一个钩子。
但是,我怀疑这是否是正确的方向。在计算完成时,默认情况下会打印对象检测指标(即AP),并且似乎是在TF图之外计算的。我不能从那里转储它,因为在该调用期间无法访问global_step和model_dir。