一种加入注意力机制的YOLOv5神经网络车辆检测方法技术

本发明专利技术公开了一种加入注意力机制的YOLOv5神经网络车辆检测方法，包括：采用汽车图像数据集，对所述数据集进行预处理；对YOLOv5神经网络进行改进，添加注意力机制；将标注好的所述数据集按照符合网络要求的格式输入所述改进后的YOLOv5神经网络进行训练并测试结果；将训练好的模型部署到移动端进行目标车辆的检测与识别。本发明专利技术提高了模型对小目标识别的效果，提升了模型的识别精度与模型收敛的速度，实现移动端对目标实时的识别功能。实现移动端对目标实时的识别功能。实现移动端对目标实时的识别功能。

【技术实现步骤摘要】
一种加入注意力机制的YOLOv5神经网络车辆检测方法


[0001]本专利技术涉及人工智能与交通安全中的图像检测与识别的
，尤其涉及一种加入注意力机制的YOLOv5神经网络车辆检测方法。

技术介绍

[0002]近年来随着科技的不断进步，人工智能技术在许多领域的应用取得了有益效果，图像的智能检测与识别技术也应运而生，随着YOLO、Tensorflow等一系列神经网络框架的设计开发，此项技术也日益成熟，若将这种图像识别技术应用在交通安全系统中，当部分路段出现道路拥挤时，可以快速对拥挤地段采取相应疏通措施，当有救护车或消防车等紧急车辆需要通行时，可以提前疏通道路以防出现紧急车辆无法通行的情况。
[0003]由于道路环境复杂，多变的背景环境会影响图像识别的准确率，现有技术采用的图像识别检测方法会受到环境的影响，没有针对性的对所训练图像的关键部分分配足够多的“注意力”，有时会出现错误识别或者遗漏识别的情况，提升识别的准确率和识别效率就成为了亟需解决的重要问题。

技术实现思路

[0004]本部分的目的在于概述本专利技术的实施例的本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种加入注意力机制的YOLOv5神经网络车辆检测方法，其特征在于，包括：采用汽车图像数据集，对所述数据集进行预处理；对YOLOv5神经网络进行改进，添加注意力机制；将标注好的所述数据集按照符合网络要求的格式输入所述改进后的YOLOv5神经网络进行训练并测试结果；将训练好的模型部署到移动端进行目标车辆的检测与识别。2.如权利要求1所述的加入注意力机制的YOLOv5神经网络车辆检测方法，其特征在于：所述汽车图像数据集包括UA
‑
DETRAC公开数据集、图像标注工具Labelimg。3.如权利要求2所述的加入注意力机制的YOLOv5神经网络车辆检测方法，其特征在于：所述的UA
‑
DETRAC公开数据集包括，24个不同地点拍摄的10个小时的视频；所述视频以每秒25帧的速度录制，分辨率为960*540像素。4.如权利要求1～3任一所述的加入注意力机制的YOLOv5神经网络车辆检测方法，其特征在于：对所述数据集进行预处理过程包括，利用图像标注工具Labelimg对所述图像数据集进行标注；根据COCO数据集格式将标注后的所述图像数据集保存为txt格式标注文件，并将所述数据集根据8：1的划分比例分为训练集和验证集。5.如权利要求1所述的加入注意力机制的YOLOv5神经网络车辆检测方法，其特征在于：添加注意力机制包括添加通道注意力模块：将输入的特征图，即H
×
W
×
C，分别经过基于H和W的全局最大池化层和全局平均池化，得到两个1
×1×
C的特征图；将所述特征图分别送入一个两层的神经网络，第一层的神经元个数为C/r，激活函数为ReLU，第二层的神经元个数为C；将输出的特征进行元素求和操作并激活，生成通道注意力模块，即M
c
；将M
c
与输入特征进行乘法运算，生成空间注意力模块需要的输入特征，其具体实现公式如下：如下：其中，H和W分别为输入特征图的高和宽，σ表示Sigmoid函数，F表示输入的特征，和分别表示经过所述平均池化和所述最大池化操作后的特征，W0和W1分别表示每层神经网络的权重函数。6.如权利要求1所述的加入注意力机制的YOLOv5神经网络车辆检测方法，其特征在于：添加注意力机制包括空间注意力模块：将所述通道注意力模块输出的特征作为所述空间注意力模块的输入特征图；所述输入特征图通过所述最大池化层和所述平均池化层，得到两个H*W*1的特征图；将所述两个特征图进行concat融合并通过一个7*7的卷积层，降维为一个channel，并经过Sigmoid函数生成空间注意力模块即M
s
；将M
s
与空间...

【专利技术属性】
技术研发人员：夏长权，汪李超，时壮壮，朱颖，徐思韵，
申请(专利权)人：扬州大学，
类型：发明
国别省市：