基于Yolov5的交通信号灯智能识别程序开发  

Development of intelligent recognition program for traffic signal lamps based on Yolov5

作　　者：郑国荣[1] 张尊栋[1] 赵文芊 柏卓茁 贾菲儿 ZHENG Guo-rong;ZHANG Zun-dong;ZHAO Wen-qian;BAI Zhuo-zhuo;JIA Fei-er

机构地区：[1]北方工业大学电气与控制工程学院,北京100144

出　　处：《智能城市》2024年第3期18-21,共4页Intelligent City

摘　　要：交通信号检测是智能汽车识别交通环境的一项重要辅助技术,现有的算法能够解决单一交叉口环境下的信号检测问题,但需要在十字路口的复杂交通环境中提高算法的精度和干扰可靠性。文章以one-stage目标检测算法Yolov5的应用为研究基础,实现多场景下的交通信号灯自动检测与识别,使用Labeling进行图片标注,通过镜像、裁剪、反转、等运行增强数据集,不断地调参实验与迭代模型训练,目标检测精度达到80%。Traffic signal detection is an important auxiliary technology for intelligent vehicles to identify traffic environment.Existing algorithms can solve the problem of signal detection in a single intersection environment,but it is necessary to improve the accuracy and interference reliability of the algorithm in the complex traffic environment of intersections.Based on the application of one-stage target detection algorithm Yolov5,this paper realizes the automatic detection and recognition of traffic lights in multiple scenes.Labeling is used to annotate pictures,the data set is enhanced by mirroring,cropping,reversing,etc.,and the parameter adjustment experiment and iterative model training are continuously carried out.The target detection accuracy reaches 80%.

关 键 词：目标检测 Yolov5 Labeling图片标注 模型训练 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]