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

Published 20 hours ago • verified publisher icon bubbliiiing

关于预训练权重

Open lzhhha opened this issue 3 years ago • 13 comments

您好，非常感谢您的开源代码，我学到了很多，但有一些问题： 1.我看readme里给出了两个yolo_weights.pth文件(一个百度云下载，一个可以直接下载)，这两个权重文件一样吗，都是在coco上预训练好的吗，还有backbone的预训练权重也是基于coco的吗？ 2.因为比较的方法没有加载预训练权重训练，所以我尝试在自己的数据集上不加载预训练权重进行训练，但结果非常差(不加载MAP=3，加载MAP=75，论文不加载权重MAP也能70)，基于您的这个代码您有什么好的建议，可以不加载预训练权重就达到至少不是现在这么差的效果呢？ 非常感谢！

lzhhha avatar Apr 17 '22 11:04 lzhhha

1、一样 2、论文不加载也能70？论文用了你自己的数据集吗

bubbliiiing avatar Apr 18 '22 15:04 bubbliiiing

感谢回复，是我用了他的数据集，现在分也能上来了，但是训练过程中容易梯度爆炸

lzhhha avatar Apr 18 '22 15:04 lzhhha

梯度爆炸？0 0很少见

bubbliiiing avatar Apr 18 '22 16:04 bubbliiiing

不好意思，我说错了，是nan

lzhhha avatar Apr 18 '22 16:04 lzhhha

确实很少见……不知道为什么0 0我自己训练基本没nan过，不管大目标还是小目标

bubbliiiing avatar Apr 19 '22 02:04 bubbliiiing

大佬，您这边可以提供ImageNet上的预训练权重吗

lzhhha avatar Apr 20 '22 10:04 lzhhha

…看看网盘

bubbliiiing avatar Apr 20 '22 14:04 bubbliiiing

网盘里不是COCO的预训练权重吗

lzhhha avatar Apr 21 '22 05:04 lzhhha

有主干的

bubbliiiing avatar Apr 21 '22 14:04 bubbliiiing

大佬，主干的预训练权重是基于ImageNet的吗？是官方的还是自己训练的呢？

lzhhha avatar Apr 21 '22 15:04 lzhhha

官方转的

bubbliiiing avatar Apr 22 '22 14:04 bubbliiiing

大佬，我还是想确认下，主干网络的预训练权重[darknet53_backbone_weights.pth]是在ImageNet上预训练的吧

lzhhha avatar Apr 22 '22 16:04 lzhhha

是

bubbliiiing avatar Apr 23 '22 15:04 bubbliiiing