城市轨道交通车载视觉YOLOv5轻量化目标检测研究  

Research on Lightweight Object Detection of YOLOv5 Based on Vehicle Vision in Urban Rail Transit

作　　者：谭飞刚 廖全蜜 翟聪 TAN Feigang;LIAO Quanmi;ZHAI Cong(School of Traffic and Environment,Shenzhen Institute of Information Technology,Shenzhen 518172,Guangdong China;School of Transportation,Civil Engineering&Architecture,Foshan University,Foshan 528225,Guangdong,China)

机构地区：[1]深圳信息职业技术学院交通与环境学院,广东深圳518172 [2]佛山科学技术学院交通与土木建筑学院,广东佛山528225

出　　处：《广东交通职业技术学院学报》2024年第1期45-48,80,共5页Journal of Guangdong Communication Polytechnic

基　　金：2021年广东省科技创新战略专项资金项目(编号:pdjh2021b0907);广东省普通高校青年创新人才项目(编号:2020KQNCX205,207);广东省基础与应用基础研究基金(编号:2022A1515010948)。

摘　　要：YOLO系列算法因其独特的优越性而被广泛研究,但是复杂的模型使得其在某些应用场景受到了限制。针对城市轨道交通无人列车驾驶中对前方轨行区异物入侵检测的时效性,本文以YOLOv5s算法为基础,从主干网络轻量化设计和边界框回归损失函数两方面进行优化。首先,通过引入GhostNet网络来减少网络参数提升模型轻量化。然后,以DIOU作为边界框回归损失函数,提高了模型收敛速度和精度。以COCO数据集进行试验,结果表明,所提算法在模型轻量化和检测精度上相比YOLOv5s均有所提升。YOLO series algorithm has been widely studied because of its advantages,but its complex model restricts itsapplication in some scenarios.In view of the timeliness of foreign object intrusion detection in the front track area during the driving of unmanned trains in urban rail transit,this paper,based on YOLOv5s algorithm,optimizes the lightweight design of the backbone network and the regression loss function of the boundary box.First,the GhostNet network is introduced to reduce the network parameters and improve the lightweight of the model.Then,DIOU is used as the boundary box regression loss function to improve the convergence speed and the accuracy of the model.The experimental results on COCO data set show that the proposed algorithm improves the model lightness and detection accuracy compared with YOLOv5s.

关 键 词：城市轨道交通 目标检测 YOLO 轻量化 无人列车驾驶 

分 类 号：U239.5[交通运输工程—道路与铁道工程]