文章目录
前言
CBAM是一个简单而有效的注意力模块,用于前馈卷积神经网络。给定一个中间特征图,我们的模块按顺序推断两个独立维度(通道和空间)的注意力图,然后将注意力图乘以输入特征图以进行自适应特征细化。由于 CBAM 是一个轻量级的通用模块,因此它可以无缝集成到任何 CNN 架构中,开销可以忽略不计,并且可以与基础 CNN 一起进行端到端训练。论文链接: (CBAM论文)
一、YOLOv8原始版本代码下载
源码下载地址 :链接: link
源码提取码: rpe7
二、CBAM代码
# -*- coding: utf-8 -*-
"""
@Auth : 挂科边缘
@File :CBAM.py
@IDE :PyChar