一种基于改进DETR的面向无人机图像的小目标检测方法

本发明公开了一种基于改进DETR的面向无人机图像的小目标检测方法。本发明基于端到端的注意力目标检测模型DETR构建，并提出了三个关键技术改进：双域混合编码器模块、增强的查询选择机制和知识蒸馏策略。其中，双域混合编码器模块用于整合空间域与频率域特征，通过自注意力机制仅对高层特征进行操作，同时利用双域融合模块结合低层的特征；其次，引入增强的查询选择机制，通过在多尺度特征图上生成锚框，结合扩展的交并比动态筛选高得分锚框作为查询；此外，通过知识蒸馏策略，将教师模型的知识迁移到轻量级学生模型，实现对小目标的精确检测与高效推理。本发明在小目标检测精度和计算量方面具有明显优势，可用于多种无人机图像处理任务。