面向无人机影像小目标检测的轻量化算法OA
Lightweight algorithm for small object detection in UAV images
针对现有检测模型对无人机影像中目标、特别是小目标检测不充分的问题,本文提出一种轻量化小目标检测模型PRSU-YOLO.首先,设计风车结构卷积自适应模块,增强对细微特征的提取能力;其次,构建重参数化空间-通道卷积模块,通过动态特征重建优化多尺度表征;再次,在颈部网络中嵌入小目标检测增强分支,构建高分辨率细节特征的强化路径;最后,引入尺度动态交并比损失函数,使模型能够自适应地调整边界框回归策略.本文模型在计算复杂度仅增加14.6GFLOPS的基础上,在VisDrone2019数据集上的mAP@0.5达到37.4%,较基线模型提升4.4%;在TinyPerson数据集上的精确率达到34.5%,提升3.5%.实验结果表明,该模型在显著提升检测能力的同时,有效控制了计算成本,为无人机对地观测场景下的目标检测任务提供了有效的解决方案.
To address the insufficient feature extraction of existing detection models for tiny targets,this paper proposes a lightweight small object detection model named PRSU-YOLO.First,a Pinwheel-shaped Convolutional Adaptive Module is designed to enhance the directional extraction capability of subtle features.Second,a Reparameterized Spatial-Channel Convolution Module is constructed to optimize multi-scale representation through dynamic feature reconstruction.Third,a Small Object Detection Branch is embedded in the neck network to establish an enhancement pathway for high-resolution detail features.Finally,a Scale-based Dynamic Intersection over Union loss function is introduced,enabling the model to adaptively adjust the bounding-box regression strategy.With only a 14.6 GFLOPS increase in computational complexity,the proposed model achieves an mAP@0.5 of 37.4%on the VisDrone2019,representing a significant improvement of 4.4%over the baseline.On the TinyPerson,it attained a precision of 34.5%,which is an increase of 3.5%compared to the baseline.The experimental results demonstrate that the model significantly enhances detection capability while effectively controlling computational cost,providing an effective solution for small object detection tasks in UAV-based ground observation scenarios.
罗可心;李松江;王鹏;杨华民
长春理工大学 计算机科学技术学院,吉林 长春 130022长春理工大学 计算机科学技术学院,吉林 长春 130022||吉林省大数据科学与工程联合重点实验室,吉林 长春 130022长春理工大学 计算机科学技术学院,吉林 长春 130022||吉林省网络数据库应用软件科技创新中心,吉林 长春 130022长春理工大学 计算机科学技术学院,吉林 长春 130022
信息技术与安全科学
计算机视觉无人机影像小目标检测特征增强
computer visionunmanned aerial vehicle imagerysmall object detectionfeature enhancement
《液晶与显示》 2026 (2)
253-266,14
吉林省科技创新平台建设项目(No.YDZJ202302CXJD027)Supported by Jilin Province's Science and Technology Innovation Platform Construction Project(No.YDZJ202302CXJD027)
评论