基于双流注意力和多尺度融合的绝缘子红外图像分割算法

An Insulator Infrared Image Segmentation Algorithm Based on Two-stream Attention and Multi-scale Fusion

  • 摘要: 针对绝缘子红外图像因噪声干扰严重、边缘信息复杂而导致的局部特征提取不足、漏分割和误分割率较高的问题,提出了一种基于双流注意力和多尺度融合的绝缘子红外图像分割算法。首先,在编码器中引入细节增强卷积模块,在保留全局特征提取能力的同时,进一步增强了对细节特征的捕捉能力;其次,提出了一种双流注意力机制,并将其嵌入了瓶颈结构中,有效兼顾局部与全局信息,从而显著降低了漏分割和误分割率;此外,在解码器中设计了多尺度融合模块,提升了特征信息的传递与复用效率,最大限度地缓解了梯度消失问题。实验结果表明,该模型的mIoU达到90.93%,mPA为95.05%,F1为94.94%,相较于基准模型分别提升了5.92%、5.27%和2.92%,充分验证了所提方法的有效性。

     

    Abstract: Aiming at the problems of insufficient local feature extraction, high rate of missed segmentation and mis-segmentation of insulator infrared images due to serious noise interference and complex edge information, an insulator infrared image segmentation algorithm based on dual-stream attention and multi-scale fusion is proposed. First, a detail-enhanced convolution module is introduced into the encoder, which further enhances the ability to capture detailed features while retaining the global feature extraction capability; second, a dual-stream attention mechanism is proposed and embedded into the bottleneck structure, which efficiently balances the local and global information to significantly reduce the leakage and mis-segmentation rates; furthermore, a multiscale fusion module is designed in the decoder to improve the the transfer and reuse efficiency of feature information, which minimizes the gradient vanishing problem. The experimental results show that the mIoU of the model reaches 90.93%, the mPA is 95.05%, and the F1 is 94.94%, which are improved by 5.92%, 5.27%, and 2.92%, respectively, compared with the baseline model, which fully verifies the effectiveness of the proposed method.

     

/

返回文章
返回