YOLOv8添加CBAM注意力机制（小白修改）

wsfeimo

于 2024-07-19 09:43:10 发布

阅读量1k

点赞数 3

CC 4.0 BY-SA版权

文章标签： YOLO

本文链接：https://blog.youkuaiyun.com/wsfeimo/article/details/140540093

修改

在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 =

最低0.47元/天解锁文章