[YOLO]快速用注意力优化C3、C3x模块

如何用轻量化注意力快速改进优化C3和C3x?

注意力介绍与代码

yoloV5中核心模块都在common.py中定义,模块解析都在yolo.py中;所以基本流程就是在common.py定义新模块,在yolo.py解析模块;然后模型文件使用模块。

我们以在C3x模块中增加ECA注意力为例:

(1)首先在common.py中定义注意力模块

将注意力代码放在C3x代码前面的任意位置(只要在调用之前就行)

(2)在C3x模块中调用上面定义的注意力

这里我直接复制了C3x模块代码,重新命名为C3xA。

class C3xA(C3):
    """Extends the C3 module with
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

七布隆咚锵

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值