一种基于YOLO的轻量双头自适应红外弱小检测算法的方法

本发明公开了一种基于YOLO的轻量双头自适应红外弱小检测算法的方法，包括以下步骤：S1、设计双检测头轻量级网络结构，其中包括特征提取模块和特征融合模块，以减少在卷积过程中信息的丢失，降低网络复杂度并提高运行效率；S2、计算各滤波器之间的相似度，根据相似度结果裁剪冗余滤波器，并调整剪枝率，以在保持核心特征信息的同时提高检测速度；S3、引入目标尺寸自适应惩罚因子和角度平衡因子，设计cosPIoU‑Varifocal损失函数，以均衡正负样本损失及重叠区域损失。本发明通过重新设计特征提取模块，提出了双检测头E‑YOLO网络结构，有效减少了多次卷积过程中信息丢失，提升了对红外弱小目标的检测能力。