面向暗光场景的目标偏振/可见光融合检测方法  

Target Polarization/Visible Light Fusion Detection Method for Dark Light Scenes

作　　者：马如钺 王晨光[1] 曹慧亮 申冲 唐军 刘俊 MA Ruyue;WANG Chengaung;CAO Huiliang;SHEN Chong;TANG Jun;LIU Jun(School of Information and Communication Engineering,North University of China,Taiyuan 030051,China;School of Instrument and Electronics,North University of China,Taiyuan 030051,China)

机构地区：[1]中北大学信息与通信工程学院,太原030051 [2]中北大学仪器与电子学院,太原030051

出　　处：《计算机测量与控制》2024年第4期46-53,共8页Computer Measurement &Control

基　　金：国家自然科学基金(61973281,51821003,51922009);山西省重点研发计划项目(202003D111003);山西省优秀青年培育项目(202103021222011);光电信息控制和安全技术重点实验室基金(2021JCJQLB055010);山西省量子传感与精密测量重点实验室(201905D121001);山西省1331工程项目。

摘　　要：为解决偏振暗光场景下常见目标识别结果准确性不高的问题,提出了基于卷积神经网络的偏振度图像与可见光图像融合算法,设计了新的损失函数以形成无监督学习过程,引入拉普拉斯算子提高融合图像的质量,最终将被测目标的偏振信息与可见光信息有效结合;提出了基于改进的YOLOv5算法对融合后的目标进行目标检测,在网络框架中加入CA注意力机制,将通道注意力机制与空间注意力机制相结合;在自制的数据集上对提出的网络进行训练测试,结果表明,融合图像在主客观上都达到了较好的视觉效果,将改进的YOLOv5算法相比最优的YOLOv5s模型,精确率和召回率分别达到了89.3%和82.5%,均值平均精度分别提高了2.6%和1.8%。In order to solve the low accuracy of common target recognition results in polarized dark light scenes,this paper proposes a fusion algorithm based on convolutional neural network of polarization degree images and visible light images,designs a new loss function to form an unsupervised learning process,introduces Laplace operator to improve the quality of fused image,and finally the polarization information of the target to be measured is effectively combined with the visible light information;A fused target detection algorithm is proposed based on the improved YOLOv5 algorithm for the target detection of the fused target,the coordinate attention(CA)attention mechanism of the network framework is added to combine the channel attention mechanism with the spatial attention mechanism.The proposed network is trained and tested on a homemade dataset,the results show that the fused image achieves better visual effects subjectively and objectively,compared with the optimal YOLOv5s model,the precision and recall rate of the improved YOLOv5 algorithm reach 89.3%and 82.5%,respectively,and the mean average precisions increase by 2.6%and 1.8%,respectively.

关 键 词：偏振成像 目标检测 图像融合 YOLOv5 图像增强 注意力机制 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]