带孔空间金字塔池化(ASPP)

  • 在DeepLab中,采用空间金字塔池化模块来进一步提取多尺度信息,这里是采用不同rate的空洞卷积来实现这一点。

  • ASPP模块主要包含以下几个部分:

    • (1)一个1×1卷积层,以及三个3x3的空洞卷积,对于output_stride=16,其rate为(6, 12, 18) ,若output_stride=8,rate加倍(这些卷积层的输出channel数均为256,并且含有BN层);

    • (2)一个全局平均池化层得到image-level特征,然后送入1x1卷积层(输出256个channel),并双线性插值到原始大小;

    • (3)将(1)和(2)得到的4个不同尺度的特征在channel维度concat在一起,然后送入1x1的卷积进行融合并得到256-channel的新特征。

    • 带孔空间金字塔池化(ASPP)_第1张图片

  • DeepLab-LargeFOV(左:即仅单个atrous conv),DeepLab-ASPP(右,即ASPP):

    • 带孔空间金字塔池化(ASPP)_第2张图片

你可能感兴趣的:(语义分割,深度学习,pytorch,神经网络)