基于全链路多尺度融合网络的小目标检测方法

作者:张铭津; 张晓雨; 杨瀚迪; 岳珂; 郭杰; 李云松; 高新波
来源:2023-09-28, 中国, CN202311275072.2.

摘要

本发明提出了一种基于全链路多尺度融合网络的小目标检测方法,实现步骤为:获取训练样本集和测试样本集;构建包括以顺次级联的注意力导向特征交叉聚合编码器、解码器和Head块为全链路,以及以连接在解码器的输入端与Head块的输入端之间的合作超分辨率块为分支网络的小目标检测模型;初始化参数;对小目标检测模型进行训练;对小目标检测模型的参数进行更新;获取小目标检测结果。本发明中的多尺度融合网络将深度加权多尺度注意力块生成的具有多尺度特征信息的特征图与跨层特征融合块生成的不同层次的特征图进行融合,并通过Head块获取高分辨率多尺度融合特征图,能够充分挖掘小目标图像的全尺度信息,有效提高小目标检测的准确率。