雷海卫, 何方圆, 贾博慧, 吴倩. 基于注意力机制的全景分割网络[J]. 微电子学与计算机, 2022, 39(1): 39-45. DOI: 10.19304/J.ISSN1000-7180.2021.0263
引用本文: 雷海卫, 何方圆, 贾博慧, 吴倩. 基于注意力机制的全景分割网络[J]. 微电子学与计算机, 2022, 39(1): 39-45. DOI: 10.19304/J.ISSN1000-7180.2021.0263
LEI Haiwei, HE Fangyuan, JIA Bohui, WU Qian. Attention-guided panoptic segmentation network[J]. Microelectronics & Computer, 2022, 39(1): 39-45. DOI: 10.19304/J.ISSN1000-7180.2021.0263
Citation: LEI Haiwei, HE Fangyuan, JIA Bohui, WU Qian. Attention-guided panoptic segmentation network[J]. Microelectronics & Computer, 2022, 39(1): 39-45. DOI: 10.19304/J.ISSN1000-7180.2021.0263

基于注意力机制的全景分割网络

Attention-guided panoptic segmentation network

  • 摘要: 针对全景分割的背景类实例重叠问题,提出了一种基于注意力机制的全景分割网络(Attention-guided Panoptic Segmentation Network, APSNet)。APSNet基于UPSNet网络进行改进.首先,在残差网络和特征金字塔之间添加一种可以同时在空间注意力和通道注意力上建模的三重态注意力机制,通过学习通道和空间特征信息,增强特征金字塔的特征提取能力;其次,在语义分割部分中,添加一种语义增强注意力机制,通过聚合多层次特征和学习空间特征信息,提升语义分割对于背景和前景的分割效果.对比实验表明,APSNet的全景质量提升了0.8%,实例级全景质量提升了2.7%,同时可以在一定程度上降低全景融合时实例分割与语义分割所发生背景类实例重叠的概率.

     

    Abstract: Aiming at the overlapping phenomenon of background instances in panoptic segmentation, an attention-guided panoptic segmentation network (APSNet) based on the attention mechanism is proposed. APSNet is improved based on the UPSNet network. Firstly, a triplet attention mechanism is added between residual network and feature pyramid network, which can model spatial attention and channel attention at the same time. By learning channel and spatial feature information, the feature extraction ability of feature pyramid network is enhanced. Secondly, in the semantic segmentation part, a kind of semantic is added to enhance attention. By aggregating multi-level features and learning spatial feature information, the segmentation effect of semantic segmentation for background and foreground is improved. The comparison experiments show that the panoptic quality of APSNet is improved by 0.8%, and the instance level panoptic quality is improved by 2.7%. At the same time, it can reduce the probability of overlapping of background instances in panoptic fusion.

     

/

返回文章
返回