基于改进YOLOv5s的红外小目标检测系统、方法以及轻量化方法

本发明公开了基于改进YOLOv5s的红外小目标检测系统、方法以及轻量化方法，涉及红外目标检测技术领域。本发明提出了一种基于改进YOLOv5s的轻量化红外小目标检测算法，也即YOLOv5s‑SIT(SIT，Small Infrared Target)。其中，所设计的YOLOv5s‑SIT算法网络中主要改进包括增强重要特征抑制非重要特征的ECBAM注意力模块、丰富特征图表达能力的spp_x融合模块，以及检测锚框的重新设置和为算法重构的损失函数ComLoss。为提高YOLOv5s‑SIT算法对红外小目标检测任务的精度，针对该算法的结构特性设计了模型轻量化策略，并对其主干网络进行了相应剪枝与压缩，最终得到轻量化红外小目标检测算法。