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

    公开(公告)号:CN120032277A

    公开(公告)日:2025-05-23

    申请号:CN202510102214.8

    申请日:2025-01-22

    Applicant: 复旦大学

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

Patent Agency Ranking