计算机工程

作者：张永亮, 陆阳, 朱芜强, 卫星, 魏臻

单位：1. 合肥工业大学 计算机与信息学院, 合肥 230009;2. 矿山物联网与安全监控技术安徽省重点实验室, 合肥 230088;3. 安全关键工业测控技术教育部工程研究中心, 合肥 230009;4. 合肥工业大学 智能制造研究院, 合肥 230009

关键词：交通标志检测,自动驾驶,卷积神经网络,目标检测,深度学习

基金：国家重点研发计划专项（2018YFC0604404）；中央高校基本科研业务费专项资金（PA2021GDGP0061）；安徽省重点研发计划项目（202004a05020040）；合肥工业大学智能制造技术研究院智能网联及新能源汽车专项（IMIWL2019003）。

基于卷积神经网络的交通标志检测算法在对现实中复杂的交通场景图像进行交通标志检测时，难以同时解决定位和分类两项任务，并且目标检测领域相关算法所使用的公开数据集提供的图像和交通标志的种类不能满足现实交通场景中复杂的情况。建立一个新的道路交通标志数据集，在YOLOv4算法的基础上针对现实交通场景图像的复杂性和图像中交通标志尺寸差异较大的特点，设计多尺寸特征提取模块和增强特征融合模块，提高算法同时定位和分类交通标志的能力。在此基础上，对算法中不同的模块设置不同的参数进行对照实验，得到一组表现最优的参数，用于检测现实交通场景图片中的交通标志。在道路交通标志数据集上的实验结果表明，该算法相比基于卷积神经网络的同类型任务目标检测算法具有更高的检测精度，平均精度均值达到83.63%。

来源：2022年第10期

《计算机工程》期刊编辑部

查看计算机工程杂志2022年第10期