yolov4训练自己的数据集

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的锅,换一个就会好很多。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值