一种基于Transformer的小目标检测方法、系统、存储介质及产品

    公开(公告)号:CN116091890B

    公开(公告)日:2024-09-27

    申请号:CN202310004108.7

    申请日:2023-01-03

    申请人: 重庆大学

    摘要: 本发明公开了一种基于Transformer的小目标检测方法、系统、存储介质及产品,首先,将待预测图像的数据集图像通过三分支vision transformer的特征提取网络,提取图像的多尺度特征;其次,采用基于注意力机制的特征融合方法,对提取出的多尺度特征进行融合,以此增强多尺度特征对图像内容的表达能力;最后,将融合的特征输入改进后的transformer decoder模块,以此进行目标检测。该方法不仅可以解决传统方法无法有效检测小目标的问题,还可以弥补部分基于Transformer的方法所出现的收敛速度慢的情况。