yolov4-pytorch icon indicating copy to clipboard operation
yolov4-pytorch copied to clipboard
verified publisher icon bubbliiiing

预训练模型问题

Open PengPeng-JunJun opened this issue 3 years ago • 5 comments

大佬，学习了您的yolo代码，在yolo5及以上版本，权重文件没有分coco和voc了，是因为两个可以共用一个预训练模型文件么？

PengPeng-JunJun avatar Feb 24 '23 09:02 PengPeng-JunJun

鹅，本身就不分，预训练都可以的

bubbliiiing avatar Feb 27 '23 14:02 bubbliiiing

我看了一下，好像只有yolo4分了voc和coco的主干提取网络的pth，其他的YOLO版本没有区分。相同的数据集，用yolo8的效果很好，但是用yolo4的时候，误用coco的预训练pth放在了voc数据集上，200个e还是没有收敛。请问这个是什么原因呢？

PengPeng-JunJun avatar Apr 06 '23 13:04 PengPeng-JunJun

额……这个预训练再voc不会不收敛的，估计是你操作的问题

bubbliiiing avatar Apr 07 '23 14:04 bubbliiiing

你好，问题找到了，不是操作的问题，是因为数据集不大，只有1000个样本，优化器选SGD，改成adma就很快收敛了

PengPeng-JunJun avatar Apr 09 '23 14:04 PengPeng-JunJun

也是再issuse中大海捞针找到了您和另一位师兄的讨论，才想到的

PengPeng-JunJun avatar Apr 09 '23 14:04 PengPeng-JunJun