今天学习的是unet unet++ rcnnunet rcnnunet注意力机制等方面的图像分割 语义分割的方法

今天学习的是unet unet++ rcnnunet rcnnunet注意力机制等方面的图像分割 语义分割的方法

下面的图代表了unet的网络架构 可以由原来的图像 转化为如下的架构，实现了网络端到端的检测

这个代表了咱们带有注意力机制的分割模型

在每一个绿色的旁边都加一个注意力的模块

注意力的代码为

class Attention_block(nn.Module):
    """
    Attention Block
    """

    def __init__(self, F_g, F_l, F_int):
        super(Attention_block, self).__init__()

        self.W_g = nn.Sequential(
            nn.Conv2d(F_l, F_int, kernel_size=1, stride=1, padding=0, bias=True),
            nn.BatchNorm2d(F_int)
        )

        self.W_x = nn.Sequential(
            nn.Conv2d(F_g, F_int, kernel_s