yolov7添加CBAM模块debug:TypeError: __init__() missing 1 required positional argument: ‘c2‘

bug:

TypeError: __init__() missing 1 required positional argument: 'c2'

我改进yolov7参考的是:

改进YOLOv7系列:23.YOLOv7添加CBAM注意力机制_yolov7 cbam-CSDN博客

YOLOv7改进:CBAM注意力机制_cbam注意力机制改进-CSDN博客

这两篇的yaml中还用到了CNeB模块和C3C2模块,于是我又参考了:

YOLOv7改进:ConvNeXt(backbone改为CNeB)_yolov7的改进bakebone-CSDN博客

解决办法:

在yolo.py的那个if中添加了C3C2和CNeB模块,如下图。

总之就是把yaml中用到的自己定义的模块(除了CBAM)加进去。

不懂具体原因,但确实解决了问题。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值