摘要
本发明公开一种红外目标检测方法、装置、设备及介质,属于人工智能技术领域。其中红外目标检测方法包括:将红外图像输入至目标检测模型的编码器模块,得到编码特征图,所述目标检测模型为预先训练好的卷积神经网络模型;将所述编码特征图输入至目标检测模型的解码器模块,获取红外图像中待检测目标的掩码;其中,编码器模块包括局部对比增强单元。本发明提出的红外目标检测方法通过计算红外图像中像素点的局部对比注意力权重,引导神经网络在编码过程中强化对潜在红外小目标空间位置的关注,并利用通道注意力强化单元实现对编码特征图的不同通道间信息的有效融合,提高对红外小目标的检测精度。
技术关键词
编码特征
编码器模块
注意力
强化特征
像素点
卷积神经网络模型
图像
加权特征
解码器
通道
存储计算机程序
人工智能技术
元素
计算机设备
矩阵
可读存储介质
上采样
处理器