bubbliiiing/yolov4-pytorch

预训练模型问题

Opened this issue · 5 comments

PengPeng-JunJun commented

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

bubbliiiing commented

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

PengPeng-JunJun commented

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

bubbliiiing commented

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

PengPeng-JunJun commented

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

PengPeng-JunJun commented

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