基于多尺度特征融合的红外小目标检测和分类方法

    公开(公告)号:CN118038152A

    公开(公告)日:2024-05-14

    申请号:CN202410187005.3

    申请日:2024-02-20

    Abstract: 本发明公开了基于多尺度特征融合的红外小目标检测和分类方法。首先根据红外小目标检测识别的任务目标,构建三分类红外图像小目标数据集;然后基于YOLOv5的先验知识设计用于红外目标检测的深度神经网络的拓扑结构Infra‑YOLO;最后基于自建的三分类红外图像小目标数据集,对提出的网络架构进行训练,获得单帧图像检测识别的模型。本发明方法能够有效提取红外图像中不同尺度大小目标的关键特征;不仅通过检测结果可视化来显示单帧红外图像中目标的位置信息和类别信息,而且以热力图的形式展示了特征图经过模型各模块显示的关键区域,可视化实验能更好地解释模型的决策过程,以此更好地辅助专业鉴定的过程。

Patent Agency Ranking