发明公开
- 专利标题: 一种基于改进YOLOv7的单帧红外弱小目标检测方法
-
申请号: CN202311648531.7申请日: 2023-12-05
-
公开(公告)号: CN117611911A公开(公告)日: 2024-02-27
- 发明人: 葛微 , 张悦 , 李兴广 , 徐小雨 , 董朋 , 赵泊宁 , 隋议萱
- 申请人: 长春理工大学重庆研究院
- 申请人地址: 重庆市渝北区龙兴镇两江大道618号
- 专利权人: 长春理工大学重庆研究院
- 当前专利权人: 长春理工大学重庆研究院
- 当前专利权人地址: 重庆市渝北区龙兴镇两江大道618号
- 代理机构: 北京华夏博通专利事务所
- 代理商 石铁岩
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06V10/82 ; G06V10/44 ; G06V10/52 ; G06V10/80 ; G06N3/0464 ; G06N3/048 ; G06N3/08
摘要:
本发明属于红外弱小目标检测技术领域,尤其为一种基于改进YOLOv7的单帧红外弱小目标检测方法,包括以下步骤:步骤1:构建单帧红外图像数据集;步骤2:构建YOLOv7网络模型;步骤3:在主干网络中的ELAN模块后引入改进的CBAM注意力机制模块,得到ELAN‑CBAM结构,并替换所述YOLO v7模型中的ELAN结构;步骤4:在头部网络中采用空洞空间金字塔形式进行多尺度融合,同时优化损失函数;步骤5:训练步骤2至步骤4构建的基于改进YOLOv7的单帧红外弱小目标检测网络。本发明能有效增强网络对弱小目标的特征提取能力,在保证检测速度的同时,解决单帧红外图像中弱小目标检测小目标漏检、背景识别错误的问题,提升单帧红外图像中弱小目标检测的召回率、精确率、平均精度均值。