Tensorflow对象检测API-在一张图中显示用于训练和验证的损失

时间:2018-08-24 07:57:58

标签: tensorflow tensorboard

我正在玩Tensorflow Object Detection API,并在自己的数据集中训练Faster R-CNN网络。我正在检查Tensorbord的学习进度。所有指标都有,但是有没有办法在一张图上同时包含两个损失图,用于训练和验证数据?还是我必须深入研究TOD Api代码并对其进行修改?我想避免使用第二种方法,因为在每次更新API时,我都必须记住一些代码是在本地更改的。

Loss plot

1 个答案:

答案 0 :(得分:1)

该图的基础数据保存在不同的标签名称下(lossloss_1)。我相信TensorBoard本身不支持在一个图中显示不同的标签。可能会有第三方扩展程序可以做到这一点。

如果不同的模型使用相同的标签,则默认情况下将合并图形(请参见:Plot multiple graphs in one plot using Tensorboard)。