首页 > 解决方案 > 在 Tensorflow 对象检测 API 中绘制验证损失

问题描述

我正在使用 Tensorflow 对象检测 API 来检测和定位图像中的一类对象。出于这些目的,我使用预训练faster_rcnn_resnet50_coco_2018_01_28模型。

我想在训练模型后检测欠拟合/过拟合。我看到了训练损失,但在评估了 Tensorboard 之后,只显示了 mAP 和 Precision 指标,没有损失。

这是否也可以在 Tensorboard 上绘制验证损失?

标签: tensorflowdeep-learningtensorboardobject-detection-apiloss

解决方案


存在验证损失。假设您使用的是最新的 API,“loss”下的曲线是验证损失,而“loss_1/2”是训练损失。


推荐阅读