一种基于深度学习的车辆检测模型构建方法.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113989769A(43)申请公布日2022.01.28(21)申请号202111245900.9(22)申请日2021.10.26(71)申请人重庆理工大学地址400054重庆市巴南区李家沱红光大道69号(72)发明人李劲松何佳(74)专利代理机构成都东唐智宏专利代理事务所(普通合伙)51261代理人罗言刚(51)Int.Cl.G06V20/58(2022.01)G06V10/25(2022.01)G06V10/774(2022.01)G06K9/62(2022.01)G06N3/04(2006.01)权利要求书1页说明书3页附图1页(54)发明名称一种基于深度学习的车辆检测模型构建方法(57)摘要本发明公开了一种基于深度学习的车辆检测模型构建方法，涉及车辆检测技术领域。本发明通过先对目标检测数据集进行制作，使其能够进行车辆目标检测；其次对YOLOv4目标检测模型结构进行优化，引入MobileNetv3改进了YOLOv4模型的CSPDarknet53特征提取网络，并改进其损失函数；再利用阿里云服务器对模型进行训练，解决了个人计算机开发难度大等问题；再对改进模型利用检测精度和检测速度指标完成评估，在树莓派上部署了AI神经计算棒和车辆检测识别模型。本发明利用轻量级卷积神经网络改进现有目标检测算法，有效解决了树莓派用于图像计算的算力不足等问题，同时有解决了对目标车辆进行实时检测的难题，使得在算力资源有限的移动式设备上可以有效进行车辆实时检测。CN113989769ACN113989769A权利要求书1/1页1.一种基于深度学习的车辆检测模型构建方法，其特征在于：包括以下步骤：利用集中的车辆图像，组成车辆训练数据集；对车辆训练数据集的车辆图像进行标注，得到目标检测数据集；改进YOLOv4目标检测模型的特征提取网络；改进目标检测模型的损失函数；构建出车辆检测识别模型。2.根据权利要求1所述的一种基于深度学习的车辆检测模型构建方法，其特征在于：其中，利用集中的车辆图像，组成车辆训练数据集，主要包括：提取PASCALVOC数据集中车辆图像组合得到车辆训练数据集。3.根据权利要求1所述的一种基于深度学习的车辆检测模型构建方法，其特征在于：所述标注为利用labelImg工具进行标注。4.根据权利要求1所述的一种基于深度学习的车辆检测模型构建方法，其特征在于：其中，改进YOLOv4目标检测模型的特征提取网络，主要包括：用深度可分离卷积的MobileNetv3改进YOLOv4目标检测模型的CSPDarknet53特征提取网络。5.根据权利要求1所述的一种基于深度学习的车辆检测模型构建方法，其特征在于：其中，改进目标检测模型的损失函数，主要包括：去掉了损失函数中的类别置信度。6.根据权利要求1所述的一种基于深度学习的车辆检测模型构建方法，其特征在于：还包括以下步骤：在个人笔记本和/或云服务器上对车辆检测识别模型进行训练。7.根据权利要求6所述的一种基于深度学习的车辆检测模型构建方法，其特征在于：还包括以下步骤：利用检测精度和检测速度指标评估车辆检测识别模型。2CN113989769A说明书1/3页一种基于深度学习的车辆检测模型构建方法技术领域[0001]本发明属于车辆检测技术领域，特别是涉及一种基于深度学习的车辆检测模型构建方法。背景技术[0002]随着自动驾驶技术的发展，使车辆对周围环境的感知更加严格，如何对前方车辆进行有效地检测识别，一直都是车辆环境感知领域的热门。[0003]近年来，深度学习技术的进步，基于深度学习的车辆检测算法已经取代精度低和鲁棒性差的传统检测算法成为主流。[0004]然而，目前基于深度学习的车辆检测算法模型结构复杂、计算量大，在算力资源有限的移动式设备上无法进行车辆实时检测。[0005]为解决上述问题，本发明提出了相关优化方案。发明内容[0006]本发明的目的在于提供一种基于深度学习的车辆检测模型构建方法，以解决了现有的问题：目前基于深度学习的车辆检测算法模型结构复杂、计算量大，在算力资源有限的移动式设备上无法进行车辆实时检测。[0007]为解决上述技术问题，本发明是通过以下技术方案实现的：一种基于深度学习的车辆检测模型构建方法，包括以下步骤：利用集中的车辆图像，组成车辆训练数据集；对车辆训练数据集的车辆图像进行标注，得到目标检测数据集；改进YOLOv4目标检测模型的特征提取网络；改进目标检测模型的损失函数；构建出车辆检测识别模型。[0008]进一步地，其中，利用集中的车辆图像，组成车辆训练数据集，主要包括：提取PASCALVOC数据集中车辆图像组合得到车辆训练数据集。[0009]进一步地，所述标注为利用labelImg工具进行标注。[0010