首页 > 解决方案 > 使用自定义数据训练 Faster RCNN 模型时总损失的变化

问题描述

我正在研究对象检测模型来识别两个类。我在 tensorflow api 中的自定义数据集上使用 Faster RCNN。该数据集包含 20k 个图像(增强),具有两个类别。在训练模型时,损失没有适当减少,因为它达到了 100k 步。如图所示,它有很多变化。有人可以告诉我我在哪里犯错。

在此处输入图像描述

标签: tensorflowdeep-learningobject-detectionobject-detection-api

解决方案


推荐阅读