Tensorboard 不显示自定义对象检测的准确度图

Posted 2023-03-27

【中文标题】Tensorboard 不显示自定义对象检测的准确度图

【英文标题】：Tensorboard does not show accuracy graph for Custom Object Detection

【发布时间】：2019-07-04 12:06:49

【问题描述】：

我一直在使用 Tensorflow 对象检测 API（网络：SSD Mobilenet V1）训练自定义对象检测器。 Tensorboard 的屏幕截图显示了网络的准确性，但是，除了准确性之外，我还显示了一堆指标。是否需要采取任何具体步骤来使用 tensorboard 显示准确度？

我正在使用更新的 model_main.py 和以下 python 命令；

python model_main.py \
    --pipeline_config_path=training/ssd_mobilenet_v1_coco.config  \
    --model_dir=training \
    --num_train_steps=560000 \
    --num_eval_steps=3 \
    --alsologtostderr 

【参考方案1】：

你训练它多久了？ 在进行第一次评估之前，它将运行多个步骤。稍等片刻，准确率和召回率数据就会显示出来

【参考方案2】：

要在 Tensorboard 中显示精度和召回率，您应该在训练模型后运行此命令，以便训练文件夹包含已训练模型的检查点

python model_main_tf2.py \ 
    --pipeline_config_path=training/ssd_mobilenet_v1_coco.config  \
    --model_dir=training \
    --checkpoint_dir= training

此命令将在培训文件夹中生成一个名为 eval 的文件夹 并显示结果，您应该在 tensorboard 中添加 eval 文件夹 (training/eval) 的路径 ！！！此命令用于 TensorFlow 2 对象检测 API