一种无人机图像的小目标检测方法、装置、介质和设备

本发明公开了一种无人机图像的小目标检测方法、装置、介质和设备，涉及图像目标检测领域，包括在原单次多边框检测SSD算法的基础上，通过主干网络进行特征提取后增加自注意力机制，在检测头前增加注意力机制，构建改进型SSD算法；对改进型SSD算法进行训练，获取用于检测无人机图像中小目标的检测模型；将待检测的无人机图像输入到检测模型中，获得目标图像，通过检测头对目标图像进行检测。本发明将多尺度特征融合和注意力机制应用于无人机图像目标检测，将深层特征与浅层特征融合，序列化通道和空间特征，能够提高无人机图像中小目标检测精度。