公开(公告)号：CN116883873A

公开(公告)日：2023-10-13

申请号：CN202310851353.1

申请日：2023-07-12

Applicant: 哈尔滨工业大学

Inventor： 霍炬 ,  孙康健 ,  刘琪 ,  杨舜渊 ,  薛牧遥 ,  周建宝

IPC: G06V20/17 ,  G06V10/774 ,  G06V10/764 ,  G06V10/766 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本申请公开了一种面向空地应用的红外小目标检测方法，属于红外小目标检测领域，包括：获取数据集，并基于数据集对红外小目标的锚框进行更新；基于GE注意力机制构建GE模块，将GE模块嵌入至YOLOv7模型的骨干特征提取网络中；基于NWD度量构建NWD模块，将NWD模块集成至YOLOv7模型的损失函数中，得到ISTD‑YOLOv7模型；基于数据集对ISTD‑YOLOv7模型进行训练和验证；基于更新的锚框，通过训练后的ISTD‑YOLOv7模型对红外小目标进行检测。本申请提供的红外小目标检测方法在红外小目标检测中具有优越性，可应用于复杂的天时天候和地形场景下准确快速的实现红外小目标检测。