面向无人机载平台的轻量级小目标检测算法OA
Lightweight Small Target Detection Algorithm for Unmanned Aerial Vehicle-borne Platforms
针对航拍影像中微小目标检测精度受限的难题,设计了一种面向无人机平台的轻量化检测框架.具体改进包括:通过增加高分辨率检测头来保留细微特征信息;采用高效交并比(Efficient Intersection over Union,EIOU)损失函数替代传统损失函数以优化参数更新效率;构建动态稀疏注意力模块来增强特征选择能力并降低计算复杂度.基于VisDrone2019基准数据集开展的对比测试表明,改进架构在平均精度均值(mean Average Precision,mAP)上较原始YOLOv5s提升显著,其中mAP0.5(mAP@EIOU=0.5)提升8.9%.消融研究证实各改进组件对各类别目标的检测准确率和推理效率均有显著提升,该方法在保持模型轻量化的同时实现了42帧/秒(Frames Per Second,FPS)的实时处理性能,有效解决了航拍场景下微小目标的识别难题.
To address the challenge of limited detection accuracy for small targets in aerial imagery,a lightweight detection framework oriented to unmanned aerial vehicle platforms is designed.Specific improvements are as follows:a high-resolution detection head is added to retain fine-grained feature information;the traditional loss function is replaced by an Efficient Intersection over Union(EIOU)loss function to optimize the efficiency of parameter update;and a dynamic sparse attention module is constructed to enhance feature selection capability while reducing computational complexity.Comparative tests conducted on the VisDrone2019 benchmark dataset show that the improved architecture achieves a significant improvement in mean Average Precision(mAP)metrics compared with the original YOLOv5s,with mAP@EIOU=0.5(mAP50)increased by 8.9%.Ablation study shows that each component significantly improved the detection accuracy and inference efficiency for all categories of targets.The optimized framework maintains the lightweight property of the model,and real-time processing performance of 42 Frames Per Second(FPS)is achieved,thus effectively solving the problem of small target recognition in aerial photography scenarios.
周侃;刘恒松;高学攀;温博;张刚;任志国
中国电子科技集团公司第五十四研究所,河北 石家庄 050081中国电子科技集团公司第五十四研究所,河北 石家庄 050081||河北省智能化信息感知与处理重点实验室,河北 石家庄 050081中国电子科技集团公司第五十四研究所,河北 石家庄 050081||河北省智能化信息感知与处理重点实验室,河北 石家庄 050081||天津大学智能与计算学部,天津 300350中国电子科技集团公司第五十四研究所,河北 石家庄 050081||河北省智能化信息感知与处理重点实验室,河北 石家庄 050081中国电子科技集团公司第五十四研究所,河北 石家庄 050081||河北省智能化信息感知与处理重点实验室,河北 石家庄 050081中国电子科技集团公司第五十四研究所,河北 石家庄 050081||河北省智能化信息感知与处理重点实验室,河北 石家庄 050081
航空航天
无人机小目标检测YOLOv5s轻量级
unmanned aerial vehiclesmall target detectionYOLOv5slightweight
《无线电工程》 2026 (2)
319-325,7
评论