小编Vol*_*mpa的帖子

如何在使用 model_main 进行训练的同时连续不断地评估 tensorflow 对象检测模型

我使用train.py和成功地训练了带有自定义示例的对象检测模型eval.py。并行运行这两个程序,我能够在训练期间在 tensorboard 中可视化训练和评估指标。

然而,这两个程序都被移到了legacy文件夹中,并且model_main.py似乎是运行培训和评估的首选方式(通过只执行一个进程)。但是,当我从model_main.py以下内容开始时pipeline.config:

train_config {
  batch_size: 1
  num_steps: 40000
  ...
}
eval_config {
  # entire evaluation set
  num_examples: 821
  # for continuous evaluation
  max_evals: 0
  ...
}
Run Code Online (Sandbox Code Playgroud)

我看到启用 INFO 日志记录model_main.py训练和评估的输出是按顺序执行的(而不是像以前那样同时执行两个进程),并且在每个训练步骤之后都会进行完整的评估。

INFO:tensorflow:Saving 'checkpoint_path' summary for global step 35932: ...
INFO:tensorflow:Saving checkpoints for 35933 into ...
INFO:tensorflow:Calling model_fn.
...
INFO:tensorflow:Done calling model_fn.
INFO:tensorflow:Starting evaluation at 2018-08-30-10:06:47
...
INFO:tensorflow:Restoring parameters from .../model.ckpt-35933
INFO:tensorflow:Running local_init_op.
INFO:tensorflow:Done running local_init_op.
INFO:tensorflow:Evaluation …
Run Code Online (Sandbox Code Playgroud)

object-detection tensorflow

6
推荐指数
1
解决办法
2776
查看次数

标签 统计

object-detection ×1

tensorflow ×1