0

所以我正在尝试制作基于Keras RetinaNet 和 ResNet-152 骨干网的对象检测模型。我已经按照每个教程解释了如何做到这一点。到目前为止,我已经训练了我的模型,我的损失是 0.81,我认为这是一个很好的结果。

分类丢失

回归丢失

这是我在标签 img 中所做的边界框过程的示例

但不知何故,当我试图预测新图像(4张图像)的结果时,结果真的很奇怪,这是每个图像的结果

文件名、框、乐谱、 标签

我在创建数据集时做错了吗?因为我只调整了一些代码,比如

  • 我评论“setup_gpu(gpu)”命令
  • 我在train.py参数部分将默认主干从“resnet50”更改为“resnet152”

对于代码,我使用了https://github.com/fizyr/keras-retinanet

编辑 :

这些是我尝试使用自己的训练数据进行预测时的示例。虽然其中一个是正确的,但是置信度分数很低,甚至没有达到 0.5。

  1. 预测 - 我
  2. 预测 - II
4

0 回答 0