UNet 改进(36):引入ELAN-FR模块(通道注意力+空间注意力)
1.创新点分析
今天我要介绍一个结合了通道注意力(Channel Attention)和空间注意力(Spatial Attention)机制的改进版UNet网络。这个网络在传统UNet架构的基础上,引入了ELAN-FR模块,显著提升了特征提取能力。
网络架构概述
这个网络主要由以下几个关键组件构成:
- 通道注意力模块(ChannelAttention)
- 空间注意力模块(SpatialAttention)
- ELAN-FR模块(融合了多尺度特征和注意力机制)
- 改进的UNet架构
核心组件详解
1. 通道注意力模块(ChannelAttention)
class ChannelAttention(nn.Module):def __init__(self, in_channels, reduction_ratio=16):super(ChannelAttention, self).__init__()self.avg_pool =