mobilenet-yolov4-pytorch icon indicating copy to clipboard operation
mobilenet-yolov4-pytorch copied to clipboard

用yolov4_resnet50网络训练自己的数据集一直没有结果

Open Tristesse-stk opened this issue 2 years ago • 2 comments

如题,确认过2007_train信息没问题,xmin,ymin,xmax,ymin,class_id排布,loss一直在下降,最后稳定在0.13左右,但是预测的时候就是什么结果也没有,得分阈值为0.5。 但是同样的数据集用yolov4-tiny的项目都能出结果。

我看了之前泡泡老师许多yolo系列的项目的Issues,可能是我的数据集不够大,而且框也不够大,但是理论上不至于比yolov4-tiny还要差。

目前我已经把Sgd改成了Adam,学习率也改了好几次,不知道还有什么可以改进的空间。

Tristesse-stk avatar Dec 14 '22 08:12 Tristesse-stk

你改大epoch

bubbliiiing avatar Dec 18 '22 13:12 bubbliiiing

跑的时候map一直为0?

bubbliiiing avatar Dec 18 '22 13:12 bubbliiiing