-
公开(公告)号:CN114283082A
公开(公告)日:2022-04-05
申请号:CN202111553869.5
申请日:2021-12-17
Applicant: 重庆邮电大学
IPC: G06T5/00 , G06N3/04 , G06V10/774 , G06V10/82
Abstract: 本发明属于图像处理与计算机视觉技术领域,特别涉及一种基于注意力机制的红外小目标检测方法,包括将高斯噪声图像和全零图像分别与原图进行通道堆叠;将通道堆叠图像传入注意力机制模块进行背景抑制和目标增强;将注意机制模块的输出图像与原图进行通道堆叠后传入红外小目标检测模块,得到红外小目标的检测图像;将检测图像与标签图像的差异作为损失函数反向优化红外小目标检测模块的网络参数;若损失函数降低到可接受的阈值范围,完成训练;将待检测图像与全零图像进行通道堆叠作为测试样本传入注意力机制模块,将注意机制模块的输出图像与原图进行通道堆叠使用红外小目标检测模块实现检测任务;本发明能够有效地抑制背景杂波,实现小目标检测。