一种基于多头自注意力机制的无人机变尺寸目标检测方法,属于目标检测技术领域,本发明为解决现有无人机目标检测算法对于小目标检测性能差的问题。本发明包括:建立数据集;建立网络结构:采用多头自注意力机制建立多头自注意力目标检测头网络,所述多头自注意力目标检测头网络的后端采用Faster Rcnn基本框架,在多头自注意力目标检测头网络的回归层再次引入自注意力机制;分步骤进行网络训练;对目标物体进行检测:对图像进行预处理后输入多头自注意力目标检测头网络,多头自注意力目标检测头网络输出检测结果。本发明用于对大小变化目标无人机的目标检测。
声明:
“一种基于多头自注意力机制的无人机变尺寸目标检测方法” 该技术专利(论文)所有权利归属于技术(论文)所有人。仅供学习研究,如用于商业用途,请联系该技术所有人。
我是此专利(论文)的发明人(作者)