计算机工程与应用2024，Vol.60Issue(13)：361-368,8.DOI:10.3778/j.issn.1002-8331.2305-0091
面向嵌入式端的轻量级交通信号灯检测算法
Lightweight Traffic Signal Light Detection Algorithm for Embedded Terminal
摘要
Abstract
In view of the existing traffic signal detection algorithms,which require a large amount of computation and model,are difficult to deploy at the embedded end,and are difficult to detect long-distance traffic signals with a high miss rate,this paper designs a lightweight traffic signal detection algorithm for the embedded end,which aims at lightweight and real-time requirements.GhostNet network Ghost module and Ghost bottleneck layer structure are adopted to reduce the number of model parameters and improve the detection speed.To solve the feature similarity problem,the weighted bidi-rectional feature pyramid network is used to make the algorithm more sensitive to the target.The dense void space pyramid pool is used to optimize the extraction of global context information.Aiming at the problem of small target recognition,the information extraction of small target is enhanced through the improvement of multi-scale detection.Finally,the learning ability of the model is improved through knowledge distillation,and the detection performance is improved.Experimental results show that the recognition accuracy and recall rate of traffic lights by this detection algorithm reach 97.0% and 99% respectively,2.7 and 3 percentage points higher than that of YOLOv5s.The model size is reduced to 8.06 MB,58% of that of YOLOv5s,and the recognition rate is increased from 51 frames per second to 56 frames per second.Through testing in the embedded terminal,the improved algorithm can recognize the traffic signal in real time and accurately.关键词
目标检测/轻量级/GhostNet/知识蒸馏/密集空洞空间金字塔池化Key words
object detection/lightweight/GhostNet/knowledge of distillation/dense atrous spatial pyramid pooling分类
信息技术与安全科学引用本文复制引用
杨永波,李栋,房建东,董祥,李毅伟..面向嵌入式端的轻量级交通信号灯检测算法[J].计算机工程与应用,2024,60(13):361-368,8.基金项目
内蒙古自治区自然科学基金(2022QN06004,2019MS06023) （2022QN06004,2019MS06023）
内蒙古自治区科技攻关项目(2019GG334,2019GG337,2019GG376). （2019GG334,2019GG337,2019GG376）
