今天学习的是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=