yolov4训练自己的数据集
网上有很多教程,试了好几个总是出现各种问题。推荐一位博主的帖子,按照他的帖子来,超级顺畅。
版权声明:本文为CSDN博主「是王同学呀」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/wxd1233/article/details/117781437
主要流程参考他的就够了,分享以下yolov4的weight:
链接:https://pan.baidu.com/s/1iry12JuOCtXezL_s06YXNQ
提取码:6666
然后需要修改以下cfg中的参数,主要参考这个:
版权声明:本文为CSDN博主「大黑山修道」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/sazass/article/details/107161088
修改classes和filters=(5+类别数)*3,分别是在3个yolo层和这3个层上一层中修改。
如果报错cuda超出内存,batchsize改小点吧,我用的2,看到一个帖子说是mish的锅,换一个就会好很多。