改进YOLOvN架构,提升小型无人机实时检测精度与速度
YolovN-CBi: A Lightweight and Efficient Architecture for Real-Time Detection of Small UAVs
- 融合CBAM注意力模块与双向特征金字塔网络,增强小目标感知能力
- 轻量化模型在4个基准数据集上实现65.73%的mAP,比教师模型高6.51%
- 蒸馏后模型速度快82.9%,适合边缘设备部署,适用于安防与反无人机场景
无人机在民用和军事领域带来日益增长的风险,亟需高精度、实时的检测系统。由于其尺寸小、移动快、视觉对比度低,检测难度大。本文提出一种改进的YolovN架构——YolovN-CBi,引入卷积块注意力模块(CBAM)与双向特征金字塔网络(BiFPN),以提升对小目标的敏感性。构建了包含2.8万张图像的训练数据集,以及含2500张极小无人机图像的本地测试集。在四个基准数据集及本地数据集上评估,所提的Yolov5-CBi架构在速度-精度权衡上优于Yolov8和Yolov12。还提出了四种变体并采用知识蒸馏优化,使用Yolov5m-CBi作为教师模型,Yolov5n-CBi作为学生模型。蒸馏后模型在[email protected]:0.9达到0.6573,较教师模型的0.6171提升6.51%,且比基线模型快82.9%,显著提升实时检测性能。
原文摘要 · Abstract (English)
Unmanned Aerial Vehicles, commonly known as, drones pose increasing risks in civilian and defense settings, demanding accurate and real-time drone detection systems. However, detecting drones is challenging because of their small size, rapid movement, and low visual contrast. A modified architecture of YolovN called the YolovN-CBi is proposed that incorporates the Convolutional Block Attention Module (CBAM) and the Bidirectional Feature Pyramid Network (BiFPN) to improve sensitivity to small object detections. A curated training dataset consisting of 28K images is created with various flying objects and a local test dataset is collected with 2500 images consisting of very small drone objects. The proposed architecture is evaluated on four benchmark datasets, along with the local test dataset. The baseline Yolov5 and the proposed Yolov5-CBi architecture outperform newer Yolo versions, including Yolov8 and Yolov12, in the speed-accuracy trade-off for small object detection. Four other variants of the proposed CBi architecture are also proposed and evaluated, which vary in the placement and usage of CBAM and BiFPN. These variants are further distilled using knowledge distillation techniques for edge deployment, using a Yolov5m-CBi teacher and a Yolov5n-CBi student. The distilled model achieved a [email protected]:0.9 of 0.6573, representing a 6.51% improvement over the teacher's score of 0.6171, highlighting the effectiveness of the distillation process. The distilled model is 82.9% faster than the baseline model, making it more suitable for real-time drone detection. These findings highlight the effectiveness of the proposed CBi architecture, together with the distilled lightweight models in advancing efficient and accurate real-time detection of small UAVs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。