如何用轻量化注意力快速改进优化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