修改
在YOLOv8中ultralytics->nn->tasks.py文件,导入CBAM模块,在
from ultralytics.nn.modules import # 添加CBAM模块



添加
在
def parse_model(d, ch, verbose=True): # 中添加CBAM模块

添加 elif m is CBAM:
if m in {BottleneckCSP, C1, C2, C2f, C2fAttn, C3, C3TR, C3Ghost, C3x, RepC3}:
args.insert(2, n) # number of repeats
n = 1
elif m is AIFI:
args = [ch[f], *args]
elif m in {HGStem,

最低0.47元/天 解锁文章
5722

被折叠的 条评论
为什么被折叠?



