轻量化模型FDM-YOLO提升小目标检测精度,适合边缘设备实时部署。
A lightweight model FDM-YOLO for small target improvement based on YOLOv8
- 在YOLOv8基础上重构网络,加入高分辨率层并移除大目标层。
- 参数量减少38%,Map0.5从38.4%提升至42.5%,推理速度几乎不变。
- 采用轻量模块和动态上采样,兼顾精度与边缘计算效率。
小目标因像素少、背景复杂、拍摄角度多变,难以有效提取特征。大型模型虽精度高,但推理时间长,不适用于边缘设备;而低算力模型常牺牲检测精度。本文针对小目标检测,在低计算约束下提出FDM-YOLO模型。通过分析YOLOv8检测头输出,引入高分辨率层并移除大目标检测层以聚焦小目标。基于PConv设计轻量结构Fast-C2f,嵌入模型的PAN模块。为缓解轻量化导致的精度下降,引入动态上采样(Dysample)和轻量级EMA注意力机制。在Visdrone数据集上的实验表明,FDM-YOLO参数量减少38%,Map0.5由38.4%提升至42.5%,推理速度基本保持不变,验证了该方法在边缘设备上兼顾精度与效率的有效性。
原文摘要 · Abstract (English)
Small targets are particularly difficult to detect due to their low pixel count, complex backgrounds, and varying shooting angles, which make it hard for models to extract effective features. While some large-scale models offer high accuracy, their long inference times make them unsuitable for real-time deployment on edge devices. On the other hand, models designed for low computational power often suffer from poor detection accuracy. This paper focuses on small target detection and explores methods for object detection under low computational constraints. Building on the YOLOv8 model, we propose a new network architecture called FDM-YOLO. Our research includes the following key contributions: We introduce FDM-YOLO by analyzing the output of the YOLOv8 detection head. We add a highresolution layer and remove the large target detection layer to better handle small targets. Based on PConv, we propose a lightweight network structure called Fast-C2f, which is integrated into the PAN module of the model. To mitigate the accuracy loss caused by model lightweighting, we employ dynamic upsampling (Dysample) and a lightweight EMA attention mechanism.The FDM-YOLO model was validated on the Visdrone dataset, achieving a 38% reduction in parameter count and improving the Map0.5 score from 38.4% to 42.5%, all while maintaining nearly the same inference speed. This demonstrates the effectiveness of our approach in balancing accuracy and efficiency for edge device deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。