如何使用Tensorflow Object Detection API在Tensorboard中选择训练点数进行转移学习

时间:2019-05-13 15:44:18

标签: tensorflow tensorboard object-detection-api

我正在使用Tensorflow对象检测API,通过传递学习进行培训。当运行Tensorboard以可视化度量时,为每个显示的两个唯一点分别位于步骤0和最后一步(因此是一条线)。如何获取每个时期或迭代的信息?

训练在进行中,因此我假设在调用model_main.py或经过预训练的模型配置文件(在我的情况下为ssd_mobilenet_v2_coco_2018_03_29.config)中,它必须是一个参数。

python -m object_detection.model_main \
  --pipeline_config_path={PIPELINE_CONFIG_PATH} \
  --model_dir={MODEL_DIR} \
  --num_train_steps=1000 \
  --alsologtostderr)

在最初为空的MODEL_DIR文件夹中,我得到了以下信息:

  • eval_0 /
  • export /
  • 检查点
  • events.out.tfevents ....
  • graph.pbtxt
  • model.ckpt-1000.meta
  • model.ckpt-0.index
  • model.ckpt-0.meta
  • model.ckpt-1000.data-00000-of-00001
  • model.ckpt-1000.index
  • model.ckpt-0.data-00000-of-00001

0 个答案:

没有答案