一种红外小型目标检测方法及系统

本发明提供一种红外小型目标检测方法及系统，构建改进YOLOv5l网络模型，改进YOLOv5l网络模型中颈部网络包括多个依次堆叠的特征提取模块，在特征提取模块中，在第一concat单元之后添加CBAM‑IRSTD单元，在相邻两个特征提取模块中，上一特征提取模块对应的第一concat单元和下一特征提取模块对应的第一concat单元通过CBL‑DP模块连接，采集若干目标检测图像并标记红外小型目标，构建包括所有标记好的目标检测图像的数据集，使用数据集对改进YOLOv5l网络模型进行训练，将待检测图像输入到训练好的改进YOLOv5l网络模型，识别待检测图像上的红外小型目标。本发明对训练迭代的拟合效果以及检测精准度都有所提升，大幅减少整体模型参数量，提升了检测速度。