FCDN:一种新的多重注意力机制的轻量化红外弱小目标检测算法

FCDN: A New Lightweight Infrared Small and Weak Target Detection Algorithm with Multiple Attention Mechanism

  • 摘要: 针对传统目标检测框架对红外弱小目标检测能力不足、召回率低、模型参数量大的问题,提出了一种新的多重注意力机制的轻量级红外弱小目标检测算法——FCDN。首先,提出了轻量级网络结构,减少目标信息在信息流传递过程中的丢失;其次,在上采样过程中采用轻量化的上采样算子来获取更大的感受野,提升召回率;再次,采用带有多重注意力机制的检测头从不同层次来更多关注小目标信息用于目标的回归和识别;最后,采用一种新的损失函数,使用最大相似包围框度量来表征真实框和预测框的相似度。实验结果表明,在复杂背景下的红外弱小目标检测数据集上,相比于基准模型YOLOv5(You Only Look Once version 5),精确率达到99.4%,召回率提高了5%,达到94.7%,mAP50提高了4.3%,达到97.8%,参数量相比原基础模型下降约42%,模型大小被压缩至8.66 Mb。检测结果表明该算法可以满足未来对于红外弱小目标的检测需求,对于红外跟踪、预警侦察、导弹制导等实际应用具有重要价值和参考意义。

     

    Abstract: In response to the shortcomings of traditional object detection frameworks in detecting weak infrared targets, such as low recall and large model parameters, a new lightweight infrared weak target detection algorithm, FCDN, is proposed with multiple attention mechanisms. Firstly, a lightweight network structure is proposed to reduce the loss of target information during the information flow transmission process. Secondly, a lightweight upsampling operator is used during the upsampling process to obtain a larger receptive field and improve the recall rate. Thirdly, detection heads is adopted with multiple attention mechanisms again to focus more on small target information from different levels for target regression and recognition. Finally, a new loss function is applied, which uses the maximum similarity bounding box metric to characterize the similarity between the real box and the predicted box. Experimental results show that on the infrared weak target detection dataset under complex backgrounds in comparison with the benchmark model YOLOv5, the accuracy reaches 99.4%, the recall rate increases by 5%, reaching 94.7%, the mAP50 increases by 4.3%, reaching 97.8%, and the parameter quantity is reduced by about 42% compared to original basic model. The model size is compressed to 8.66 MB. The detection results show that the proposed algorithm can meet the future detection needs of infrared weak targets, which has important value and reference significance for practical applications such as infrared tracking, early warning reconnaissance, missile guidance, etc.

     

/

返回文章
返回