[pytorch]语义分割之DANet: Dual Attention Network for Scene Segmentation(训练代码+预测代码)
一,DANet(双注意力模型)网络结构:网络有两个注意力分支,一个是位置注意力模型(PAM),另一个是通道注意力模型(CAM):位置注意力模型(PAM):A是通过Resnet网络提起的特征图,维度是CxHxW。把A送入一个卷积层,得到特征图B和C,B和C的维度也是CxHxW。然后把B和C的维度reshape成CxN,(N=HxW)即N是特征图的像素个数。然后再把B进行矩阵转置,维度就变成NxC。然