YOLOv4-LVR:面向边缘设备的轻量级车辆识别算法

YOLOV4-LVR: A LIGHTWEIGHT VEHICLE RECOGNITION ALGORITHM FOR EDGE DEVICES

  • 摘要: 已有的车辆识别系统网络结构复杂、模型内存占用量大,难以部署到资源受限的边缘设备上。为此提出一种基于改进YOLOv4的轻量级车辆识别算法(YOLOv4-LVR)。替换主干网络、使用深度可分离卷积对网络进行轻量化改进,通过添加注意力机制、更新先验框、引入Focal Loss损失函数来提升网络性能。KITTI数据集上的实验表明,该算法在保持较高精度的前提下,大幅降低了网络的计算和存储开销,能够以较低的成本部署到边缘设备上,对目标车辆进行有效的识别。

     

    Abstract: The existing vehicle recognition system has a complex network structure and a large model memory footprint, making it difficult to deploy to edge devices with limited resources. To this end, a lightweight vehicle recognition algorithm called YOLOv4-LVR based on improved YOLOv4 is proposed. We replaced the backbone network and used depthwise separable convolution to make lightweight improvements to the network, and then improved network performance by adding attention mechanism, updating a priori box, and introducing Focal Loss. Experiments on the KITTI dataset show that proposed algorithm greatly reduces the computational and storage overhead of the network while maintaining high accuracy, and can be deployed on edge devices at a low cost to effectively identify target vehicles.

     

/

返回文章
返回