基于改进DeepLabv3+的电力设备红外图像分割算法

Infrared Image Segmentation Algorithm of Power Equipments Based on Improved DeepLabv3+ Algorithm

  • 摘要: 针对复杂背景下电力设备红外图像分割精度低、耗时长的问题,本文提出一种基于改进DeepLabv3+算法的电力设备红外图像分割算法,首先,使用轻量化CA-MobileNetV3代替Xception实现特征提取,减少模型参数的同时提升分割准确率;其次,用SP-Dense ASPP替换ASPP,以提取更密集、更广范围的细节特征并增强长条特性;最后,引入ECA注意力机制,实现不同层级特征信息有效融合,提升模型分割精度及鲁棒性。实验结果表明,本文算法相较于较为先进的4种语义分割模型在实际电力设备红外图像分割任务中具有更高的可行性和有效性,MPA平均提升2.67%,mIoU平均提升9.32%。

     

    Abstract: This study aims to address the problems of low accuracy and long processing times of infrared image segmentation of power equipment in complex backgrounds. In this study, an infrared image segmentation algorithm based on the improved DeepLabv3 + algorithm is proposed for power equipment. First, lightweight CA − MobileNetV3 was used instead of Xception to realize feature extraction, reduce model parameters, and improve segmentation accuracy. Second, atrous spatial pyramid pooling (ASPP) was replaced with spatial pyramid (SP)-Dense ASPP to extract a denser and wider range of detailed features and enhance the strip characteristics. Finally, the efficient channel attention (ECA) mechanism was introduced to realize the effective fusion of different levels of feature information and improve segmentation accuracy and robustness of the model. The experimental results showed that the proposed algorithm had higher feasibility and effectiveness in the actual infrared image segmentation task of power equipment than the four more advanced semantic segmentation models. The average increase in mean pixel accuracy (MPA) was 2.67%, and the average increase in mean intersection over union (mIoU) was 9.32%.

     

/

返回文章
返回