在计算机视觉领域,UNet因其优异的性能在图像分割任务中广受欢迎。
本文将详细介绍一个改进版的UNet架构,它融合了Inception模块和CBAM(Convolutional Block Attention Module)注意力机制,显著提升了模型的特征提取能力。
架构概述
这个改进的UNet架构名为UNetWithInceptionCBAM
,主要包含以下几个关键组件:
-
基础双卷积模块:标准的UNet卷积块
-
Inception模块:多尺度特征提取
-
CBAM注意力机制:通道和空间注意力
-
下采样和上采样路径:标准的UNet结构
核心组件详解
1. CBAM注意力模块
CBAM(Convolutional Block Attention M