摘要
本发明公开了一种基于龙格-库塔残差块的红外小目标检测方法,主要解决现有方法检测准确率较低的问题,实现步骤为:获取训练样本集和测试样本集;构建基于龙格-库塔残差块的红外小目标检测模型:包括顺次连接的编解码网络和Head模块,以及边缘提取网络;对红外小目标检测模型进行迭代训练;获取红外图像小目标检测结果。本发明编解码网络中的龙格-库塔残差块利用注意力机制在捕获长距离依赖关系和卷积神经网络在提取局部特征方面的优势来提取语义和保留细节,可以有效增强目标特征并抑制高频噪声,边缘提取网络能够从多个层次给予目标清晰的边缘信息,提高红外小目标检测的准确率。
- 单位