高级检索

基于改进UNet的电力设备红外图像分割模型

Infrared Image Segmentation Model for Power Equipment Based on Improved UNet

  • 摘要: 针对红外图像对比度低、背景复杂且轮廓信息模糊的特点,本文提出一种基于UNet改进的电力设备红外图像分割模型UNet++。该方法充分利用UNet的编码-解码结构,并在此基础上重新设计密集连接与跳跃路径,加深了“U”型网络的层次,并对网络模型进行剪枝处理,从而在提取更多图像细节特征的同时,避免了推理速度的下降。实验使用UNet++进行了训练与验证,结果显示其平均分割精度可达88%以上。研究表明,将图像分割技术应用于红外场景,能够减少传统人工判断设备所耗费的繁琐过程,实现对海量数据的快速且高准确率的分析,最终达成自动分割的目标。

     

    Abstract: Aiming at the characteristics of infrared images with low contrast, complex backgrounds, and blurred contour information, this paper proposes an improved infrared image segmentation model, UNet++, for power equipment based on UNet. The method makes full use of the encoding-decoding structure of UNet, on this basis, redesigns the dense connections and jump paths, and deepens the hierarchy of the “U” network. The network model is pruned so as to extract more detailed features from the image while avoiding the decrease of inference speed. The experiments are trained and validated using UNet++, and the results show that the average segmentation accuracy can reach more than 88%. The study shows that applying image segmentation technology to infrared scenes can reduce the tedious process consumed by traditional manual judgment devices, achieve fast and highly accurate analysis of massive data, and finally reach the goal of automatic segmentation.

     

/

返回文章
返回