公开(公告)号：CN118982658A

公开(公告)日：2024-11-19

申请号：CN202411188622.1

申请日：2024-08-27

Applicant: 哈尔滨理工大学

Inventor： 王微 ,  赵梓航

IPC: G06V10/25 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08 ,  G06V10/40 ,  G06N3/048 ,  G06V10/80 ,  G06V10/764 ,  G06V10/77

Abstract: 本发明涉及红外弱小目标检测技术领域，具体公开了一种基于改进YOLO v8模型的红外弱小目标检测方法。该方法包括以下步骤：首先，构建YOLO v8网络模型；然后，在主干网络中引入PConv和多尺度空洞注意力(MSDA)模块，构建P‑MSDA模块，并结合C2f结构形成P_C2f_MSDA模块，替换原YOLO v8模型中的C2f结构；接着，在头部网络中集成ODConv与EffectiveSE注意力机制，结合C2f结构构成O_E_C2f模块，替换YOLO v8模型中的C2f结构；最后，对构建的基于改进YOLO v8的红外弱小目标检测网络进行训练。本发明有效提升了网络对红外图像中弱小目标的特征提取能力，在保证检测速度的前提下，显著降低了弱小目标的漏检率和背景识别错误率，从而提高了红外图像中弱小目标检测的召回率、精确率和平均精度均值。