小数据集训练没有预测效果怎么回事？

[xing-bing]

我用模型训练的口罩数据集训练完之后检测不出来，而且置信度很低，只有0.00几， 但是同样的数据集我使用你的retinanet模型训练完之后就能检测出来， 不知道是哪里出了问题

[bubbliiiing]

先看2007_train.txt里面有没有目标信息吧

[xing-bing]

有信息的， 这个数据集我在retinanet上面跑过， 在你的retinanet网络中可以正常实现， 检测结果也挺好，但是在yolo4上面训练之后就没有检测出来，置信度很低。

[bubbliiiing]

损失多少？

[bubbliiiing]

train.py和2007_train.txt截图

[xing-bing]

0.15左右

[xing-bing]

[bubbliiiing]

数据量？

[xing-bing]

576

[xing-bing]

拆分0.1作为测试集

[bubbliiiing]

改了什么参数？

[xing-bing]

训练损失正常， 一直往下降， 到0.1几左右收敛

[xing-bing]

类文件改了， 冻结50代训练

[bubbliiiing]

没有解冻？

[xing-bing]

后50代解冻

[bubbliiiing]

sgd？

[xing-bing]

嗯嗯

[xing-bing]

conf 预测值都是0.00几， 太低了，

[bubbliiiing]

batch？

[xing-bing]

200

[xing-bing]

0.15大概就收敛了， 基本就不下降了

[bubbliiiing]

= =batch_size……………………

[bubbliiiing]

多少

[xing-bing]

[xing-bing]

8， 4

[bubbliiiing]

描述准确一些，你这个不是200代么，为什么是 冻结50代训练、后50代解冻

[xing-bing]

200代是前50代冻结， 后面解冻， 之前跑一百代的时候前50冻结， 后50解冻。

[xing-bing]

我也是搞旋转框的时候一直不行， 然后换成水平框训练一下结果水平框也不行。

[bubbliiiing]

这个训练完看看，因为你的图片数量较少，理论上更新的step很少，100 epoch是比较少的，200-300差不多，如果还是没效果，建议关闭mosaic，换adam优化器试试，yolov4检测口罩没理由检测不到，

[bubbliiiing]

那你自己改了别的代码么？

[xing-bing]

没有， 搞这个的时候重新下载的

[bubbliiiing]

建议训练完看一下，如果依然效果不好，就关闭mosaic试试。因为我看你的目标很小

[bubbliiiing]

mosaic加了更小了，本来yolo就只有416

[xing-bing]

行， 我今晚搞完再试一下

[xing-bing]

你就是我的目标检测导师

[bubbliiiing]

加油

[xing-bing]

向你学习， 大佬

[xing-bing]

老哥， 我今天又用yolo5的源码跑了一下， 可以正常检测出来， 会不会是咱这代码有点问题啊

[bubbliiiing]

u版？

[xing-bing]

嗯嗯

[xing-bing]

[bubbliiiing]

数据集可分享？我自己跑一遍

[xing-bing]

可以

[xing-bing]

咋分享

[bubbliiiing]

bubbliiiing@qq.com

[xing-bing]

已发送

[bubbliiiing]

试了一下。 sgd要求更多的更新代数，你的数据集太少，每个epoch的更新step太小，用adam效果好得多。

[bubbliiiing]

[xing-bing]

你设置了多少代啊

[bubbliiiing]

100

[xing-bing]

ok, 我再来试试