当前位置: 首页> 学位论文 >详情
原文传递 基于深度学习的道路交通标志识别算法研究
论文题名: 基于深度学习的道路交通标志识别算法研究
关键词: 道路交通标志;图像识别;深度学习;多级残差结构;特征金字塔结构
摘要: 在大数据和强劲的计算机算力加持下,人工智能技术得以迅速发展,其应用丰富了人们的生活。面对复杂多变的交通路况,为保障驾驶安全并提升驾驶体验,智能驾驶技术进入了人们的视野。作为智能驾驶技术不可或缺的一部分,开展对道路交通标志识别技术的研究具有重要意义。
  道路交通标志目标小且类型多样,同时在真实路况中存在标志牌被污染或被异物遮挡等场景,因此对于道路交通标志的识别任务很难通过传统识别算法实现。针对上述问题,本文基于深度学习,从网络结构设计和相关算法优化等方面开展研究,主要工作内容如下:
  1.道路交通标志数据集构建:针对道路交通标志公共数据集存在的问题,本文构建了包含25类的道路交通标志数据集,共计数据图像9888张,同时完成目标类别和位置信息的详细标注,共58560个真实框。通过与道路交通标志公共数据集和目标检测数据集的对比分析,验证了本文数据集的合理性和优越性。
  2.道路交通标志识别算法设计:本文提出使用高斯混合模型取代K-means聚类,提高预测框回归精确度。设计多尺度识别网络,其中通过多级残差结构增加网络深度,采用特征金字塔结构共享语义信息,通过五尺度预测提高小目标识别效果。设计新的损失函数,解决训练样本数量不均衡导致识别效果差的问题。采用几种基础训练策略,提高网络训练能力。
  3.算法整体性能测试:本文在构建的道路交通标志数据集上对本文算法进行了大量实验,实验结果表明本文算法对道路交通标志的识别准确度为95.0%,比SSD、RefineDet、YOLO v3、SPP-YOLO v3、YOLO v4、YOLO v5-x分别高22.3%、9.6%、5.7%、4.9%、4.3%、1.9%,识别速度达到23帧/秒,比Faster RCNN、SSD、RefineDet、YOLO v3、SPP-YOLO v3、YOLO v5-x分别快283.3%、91.7%、130.0%、15.0%、4.5%、27.8%。通过服务器本地对比实验,验证了本文算法的优越性。通过将本文算法部署到车载硬件平台,实现每帧用时0.04秒的快速准确识别,满足实时性要求。通过室外实验进一步表明本文算法在应用层面的可行性。
作者: 陈超
专业: 电子与通信工程
导师: 高明煜;梁汇江
授予学位: 硕士
授予学位单位: 杭州电子科技大学
学位年度: 2022
检索历史
应用推荐