基于深度学习的车辆智能检测方法研究.pdf

科学技术创新 2023.26基于深度学习的车辆智能检测方法研究冯思嘉（四川工商学院 计算机学院，四川 成都）引言随着现代化社会的发展，尤其是在进入“十四五”时期之后，工业和制造业进入了高速的发展，这也使得城市交通拥堵变得愈发严重。交通拥堵所带来的危害除了会耽误市民的出行时间之外，同时还会造成额外的交通污染和经济损失。根据中国交通部发表的数据显示，交通拥堵带来的经济损失占城市人口可支配收入的 20%,相当于每年国内生产总值损失 58%,每年达 2500 亿元人民币。实施智慧交通能够有效地缓解城市交通拥堵情况，其中交通数据是智慧交通有效管理的重要依据之一，因此对车辆进行检测，获得实时交通数据成为至关重要的环节1。近年来随着深度学习技术的飞速发展，使得目标检测技术也得到了发展，将基于深度学习的目标检测技术用于智慧交通能够有效地提高道路交通管理的效率，有助于城市交通拥堵的改善。1基于深度学习的目标检测方法1.1研究思路基于深度学习的目标检测算法分为两类：基于分类的卷积神经网络目标检测和基于回归的卷积神经网络目标检测2。基于分类的卷积神经网络目标检测算法有：R-CNN 算法3、SPP-NET 算法4以及 Fast R-CNN5，这种类型的检测算法优势在于检测速度较快。基于回归的卷积神经网络目标检测算法是 YOLO算 法6。故 论 文 以Yolov5 算 法 作 为 基 础，以UA-DETRAC 为数据集训练和验证算法，并对实验结果进行可视化处理。1.2Yolov5 算法介绍Yolov5 的 网 络 结 构 主 要 可 以 分 为 输 入 端、Backbone、Neck、Prediction 四个部分7。Yolov5 相比于Yolov3 而言，输入端中新增了 Mosaic 数据增强、自适应锚框计算、自适应图片缩放；Backbone 主要新增了Focus 结构、CSPNet 结构；Neck 中使用了 FPN 结构、PAN 结构；Prediction 中使用了 GIoU_Loss 损失函数。Yolov5 的输入端（Input）任务主要是对图像做预处理操作，它的网络模型的输入图像尺寸是 640640，因此在图像传入模型之前要将图像自适应地缩放成640640，以提高检测速度。主干网络部分（Backbone）的任务主要是对图像进行特征提取。主要流程如下：先对图片进行切片，然后进行拼接，再卷积。在切片和拼接的过程中使用到了Focus 结构。颈部网络（Neck）的主要任务是收集主干网络不同阶段的特征图，然后再将这些特征进行混合，以便摘要：随着现代社会的不断发展，车辆的数量每年都在不断增加，这使得交通拥堵变得愈发严重。智能化的交通管理能够有效缓解交通拥堵情况，其中对车辆进行检测是智慧交通实施的一重要环节。传统的车辆检测方法效率较低同时也依赖人工操作，鲁棒性较差，基于深度学习的车辆检测能够很好地解决上述问题。因此论文采用 Yolov5 算法，利用 UA-DETRAC 数据集，实现对车辆的智能检测，实验结果表明，通过训练的模型对图片中的车辆进行检测，模型准确度较高，检测结果较好，mAP 为 0.95。关键词：Yolov5；深度学习；目标检测；车辆检测中图分类号院TP391.4文献标识码院A文章编号院2096-4390渊2023冤26-0084-04作者简介：冯思嘉（1995-），女，硕士，研究方向：计算机视觉。84-2023.26 科学技术创新对特征信息进行增强。在颈部网络中包含了 FPN 结构和 PAN 结构。预测端（Prediction）的主要任务是对特征进行映射，对不同尺度的特征图进行预测。在预测端里主要结构有损失函数和非极大值抑制，使用 CIOU_Loss 函数作为预测框的损失函数，进行反向传播并且进行不断优化，最后再使用非极大值抑制对结果进行筛选。2深度学习实验建立2.1数据集与运行环境介绍实验采用 UA-DETRAC 数据集，该数据集是车辆检测和跟踪的大规模数据集，数据集主要拍摄于北京和天津的道路过街天桥（京津冀场景），并标注了 8250 个车辆和 121 万目标对象外框。车辆分为轿车、公共汽车、厢式货车和其他车辆四类。天气情况分为多云、夜间、晴天和雨天四类。Yolov5 算法在 Python 环境下进行运行，Python 环境使用 pip 严格按照 requirements.txt 中的要求进行搭建。其中 IDE 采用 PyCharm，与其他开发环境相比，pycharm 有可视化的界面，输入代码，调试都非常方便。利于调试以及大型项目开发。Numpy 依赖库用于数组快速操作的各种 API，包括数学、逻辑、形状操作、排序、选择、输入输出、离散傅立叶变换、基本线性代数，基本统计运算和随机模拟等等。pillow 依赖库是常用的图像处理库。2.2模型训练与推理过程在使用 Yolov5 进行训练时，参数如表 1 所示。表 1 训练参数在对模型进行训练之前，首先利用 labelimg 工具对数据集中的图片进行人工打标、基础数据增强以及去噪等操作。然后对模型使用上述参数进行训练，训练结束之后，使用该模型对测试集中的数据进行预测。2.3训练数据可视化与分析对模型训练期间的数据进行可视化分析，图 1 为混淆矩阵（confusion matrix），混淆矩阵是对分类问题的预测结果的总结。使用计数值汇总正确和不正确预测的数量，并按每个类进行细分，这是混淆矩阵的关键所在。混淆矩阵显示了分类模型的在进行预测时会对哪一部分产生混淆。它不仅可以让我们了解分类模型所犯的错误，更重要的是可以了解哪些错误类型正在发生。混淆矩阵的每一列代表一个类的实例预测，而每一行表示一个实际的类的实例，这样可以通过矩阵就能看出机器是否将两个不同的类混淆。图 1 的混淆矩阵可以看出经过训练之后，其结果较好。图 1混淆矩阵接着获取训练过程中的单一准确率和召回率。图2 和图 3 分别为 P_curve、R_curve。其中 P_curve 为单一类准确率，反应的是准确率和置信度的关系图。R_curve 为召回率额置信度的关系图，当置信度越小的时候，类别检测越全面。图 2 中 Precision 代表预测为 positive 的准确率，它从预测结果角度出发，描述了二分类器预测出来的正例结果中有多少是真实正例，即该二分类器预测的正例有多少是准确的，公式为：（1）其中，TP 代表真阳性，即样本的真实类别是正参数名字 参数设置 图像输入大小（images size）6406403 训练轮数（epochs）100 批次大小（batch size）32 优化器（optimizer）SGD 动量（momentum）0.937 初始学习率（lr0）0.01 学习率因子（lrf）0.2 热启动（warm up iteration）1000 权重衰减系数（weight decay）0.0005 iou 损失系数（box loss gain）0.02 cls 损失系数（cls loss gain）0.2 PrecisionTPTPFP85-科学技术创新 2023.26例，并且模型预测的结果也是正例，预测正确；FP 代表假阳性样本的真实类别是负例，但是模型将其预测成为正例，预测错误。从图 2 中可以看出 Precision 的数值随着训练的进行逐渐平稳地趋近于 1，说明经过训练模型的准确度较高。图 3 中 Recall 代表真实为 positive 的准确率，即正样本有多少被找出来了，公式为：（2）其中，FN 表示假阴性，样本的真实类别是正例，但是模型将其预测成为负例，预测错误。图 4 为训练的损失函数，损失函数是用于衡量预测模型和真实值不一样的程度，其中 box_loss 描述了预测框与标定框之间的误差从图中可以看出，随着训练地不断进行 box_loss 的值越来越小；obj_loss 描述了网络的置信度，cls_loss 用于计算锚框与对应的分类是否正确。从图 4 可以看出，随着训练的进行，box_loss、obj_loss 和 cls_loss 都趋近于 0。2.4实际检测对训练好的 Yolov5 模型的准确度进行检测，使用该模型检测，检测结果如图 5 所示。根据图 5 可以看出，该模型可以准确地检测出途中的车辆，并且将车辆用方框标注出来，同时置信度较高在 0.690.88 之间，可以说明模型训练结果较好。ecallTPRTPFN图 4训练过程损失函数图 3R_curve图 2P_curve86-2023.26 科学技术创新Research on Intelligent Detection MethodsBased on Deep LearningFeng Sijia（School of Computer,Sichuan Technology and Business University,Chengdu,China）Abstract:With the continuous development of modern society,the number of vehicles is increasing everyyear,which makes traffic congestion more and more serious.Intelligent traffic management can effectivelyalleviate traffic congestion,and vehicle detection is an important part of the implementation of intelligenttransportation.Traditional vehicle detection methods are inefficient and rely on manual operation,and therobustness is poor,vehicle detection based on deep learning can solve the above problems well.Therefore,theYolov5 algorithm is used to realize the intelligent detection of vehicles by using the UA-DETRAC dataset,and the experimental results show that the model is more accurate and the detection results are better by thetrained model,and the mAP is 0.95.Key words:Yolov5;deep learning;target detection;vehicle detection图 5车辆实际检测结果结束语论文将深度学习算法中的 Yolov5 算法应用于道路车辆检测中，利用 UA-DETRAC 数据集进行训练、测试和验证。通过分析训练过程中的数据和实际的检测效果发现，可以对道路车辆进行准确地检测，实现了道路车辆的检测，对于缓解城市交通拥堵有着一定的意义，可以提高城市对于道路车辆的效率，同时为车辆的智能驾驶奠定基础。参考文献1张学志，赵红东，刘伟娜，等.基于改进 YOLOv5 的红外车辆检测方法J/OL.红外与激光工程:1-112023-07-27.http:/ YOLOv5+DeepSort算法模型的交叉路口车辆实时检测J.计算机工程与科学，2023，45(4):674-682.3Girshick R.Fast r-cnn C.Proceedings of theIEEEinternationalconferenceoncomputervision.2015:1440-1448.4He K,Zhang X,Ren S,et al.Spatial pyramidpooling in deep convolutional networks for visual5recognitionJ.IEEEtransactionsonpatternanalysisandmachineintelligence,2015,37(9):1904-1916.6院老虎，常玉坤，刘家夫.基于改进 YOLOv5s 的雾天场景车辆检测方法J.郑州大学学报(工学版)，2023，44(3):35-41.7章程军，胡晓兵，牛洪超.基于改进 YOLOv5 的车辆目标检测研究J.四川大学学报(自然科学版)，2022，59(5):79-87.87-