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

yolov4的map不如yolov3

Open summer261 opened this issue 2 years ago • 8 comments

博主您好,使用了您的yolov3和yolov4在相同的数据集上进行了训练,yolov4使用的是VOC07+12+COCO的预训练权重,yolov3使用的是COCO-Train2017与训练权重,yolov4使用了mosaic,训练的参数都是默认的没有修改,为什么yolov4的map比yolov3低了10个点?请问这是预训练权重的问题吗?还是和anchor的大小有关?还是其他的问题?

summer261 avatar Mar 12 '22 03:03 summer261

@summer261 yolov4使用的是VOC的预训练权重?就是那个yolo4_voc_weights.pth?还是说哪个?

XuecWu avatar Mar 12 '22 03:03 XuecWu

@XuecWu yolo4_voc_weights.pth

summer261 avatar Mar 12 '22 15:03 summer261

数据集不合适吧,小目标比较少目测。

bubbliiiing avatar Mar 14 '22 16:03 bubbliiiing

@bubbliiiing 我是检测火灾的,现在3600张图,如果要添加小目标的话,大概要多少张合适? 500张够吗?

summer261 avatar Mar 15 '22 06:03 summer261

e……你可以不开mosaic试试,这个是为了增加小目标的检测能力

bubbliiiing avatar Mar 17 '22 13:03 bubbliiiing

@bubbliiiing v3不是没有mosaic嘛..
v4开不开差别不大,开了也就高一个点

summer261 avatar Mar 17 '22 14:03 summer261

我也是yolov4比yolov3低了十个点左右,用的是同样的数据集,所以是什么原因导致yolov4的精度这么低呢

lunaWU628 avatar Sep 19 '22 02:09 lunaWU628

你可以关了mosaic试试

bubbliiiing avatar Sep 19 '22 15:09 bubbliiiing