arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.06002cs.CV

基于轻量化YOLO7-tiny的快速车辆检测算法

Fast vehicle detection algorithm based on lightweight YOLO7-tiny

School of Mechanical Engineering, Hubei University of Technology(湖北工业大学机械工程学院)

Bo Li, YiHua Chen, Hao Xu, Fei Zhong

AI总结:

本文提出Ghost-YOLOv7轻量化车辆检测算法,通过Ghost卷积、Ghost-BiFPN、Ghost解耦头、坐标注意力和WIoU损失函数,在降低计算量与参数量的同时提升检测精度和速度。

摘要:

车辆的快速、精确检测在智能交通系统中具有重要作用。当前车辆检测算法面临计算复杂度高、检测率低以及在移动设备上可行性有限等挑战。为解决这些问题,本文提出一种基于YOLOv7-tiny(You Only Look Once version seven)的轻量化车辆检测算法,称为Ghost-YOLOv7。将模型宽度缩放至0.5,并将骨干网络的标准卷积替换为Ghost卷积,以实现更轻量的网络并提高检测速度;随后在颈部网络中嵌入一种自设计的Ghost双向特征金字塔网络(Ghost-BiFPN),以增强算法的特征提取能力并丰富语义信息;并采用Ghost解耦头(GDH)对车辆位置和类别进行准确预测;最后,在输出层引入坐标注意力机制以抑制环境干扰。采用WIoU损失函数进一步提升检测精度。在PASCAL VOC数据集上的消融实验结果表明,Ghost-YOLOv7优于原始YOLOv7-tiny模型。其计算量减少29.8%,参数量减少37.3%,模型权重减少35.1%,平均精度均值(mAP)提高1.1%,检测速度相比原始算法提高27FPS。Ghost-YOLOv7还在KITTI和BIT-vehicle数据集上进行了比较,结果表明该算法具有整体最佳性能。

英文摘要:

The swift and precise detection of vehicles plays a significant role in intelligent transportation systems. Current vehicle detection algorithms encounter challenges of high computational complexity, low detection rate, and limited feasibility on mobile devices. To address these issues, this paper proposes a lightweight vehicle detection algorithm based on YOLOv7-tiny (You Only Look Once version seven) called Ghost-YOLOv7. The width of model is scaled to 0.5 and the standard convolution of the backbone network is replaced with Ghost convolution to achieve a lighter network and improve the detection speed; then a self-designed Ghost bi-directional feature pyramid network (Ghost-BiFPN) is embedded into the neck network to enhance feature extraction capability of the algorithm and enriches semantic information; and a Ghost Decouoled Head (GDH) is employed for accurate prediction of vehicle location and species; finally, a coordinate attention mechanism is introduced into the output layer to suppress environmental interference. The WIoU loss function is employed to further enhance the detection accuracy. Ablation experiments results on the PASCAL VOC dataset demonstrate that Ghost-YOLOv7 outperforms the original YOLOv7-tiny model. It achieving a 29.8% reduction in computation, 37.3% reduction in the number of parameters, 35.1% reduction in model weights, 1.1% higher mean average precision (mAP), the detection speed is higher 27FPS compared with the original algorithm. Ghost-YOLOv7 was also compared on KITTI and BIT-vehicle datasets as well, and the results show that this algorithm has the overall best performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑