bubbliiiing / mobilenet-yolov4-pytorch

这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
MIT License
369 stars 104 forks source link

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

Open Tristesse-stk opened 1 year ago

Tristesse-stk commented 1 year ago

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

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

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

bubbliiiing commented 1 year ago

你改大epoch

bubbliiiing commented 1 year ago

跑的时候map一直为0?