一种基于深度学习的收费站各车道通行状态检测的方法技术

本发明专利技术公开了一种基于深度学习的收费站各车道通行状态检测的方法。首先构建收费站LED灯上文字相关的数据集，同时构建车道通行状态数据库；然后构建结合卷积注意力模块和密集连接模块的卷积神经网络分类模型，通过使用收费站LED灯上文字数据集和卷积神经网络分类模型，训练得到收费站LED灯上文字识别的模型；将待检测数据传入检测模型，可以得到该图中LED灯上的文字；根据得到的LED灯上的文字信息，可以得到该车道的通行状态。本发明专利技术的优点在于将传统的文字识别任务转化为已知类别的分类任务，大大降低了开销，进而完成对收费站车道通行状态的判断。车道通行状态的判断。车道通行状态的判断。

【技术实现步骤摘要】
一种基于深度学习的收费站各车道通行状态检测的方法


[0001]本专利技术涉及图像识别与机器视觉
，具体涉及一种基于深度学习的收费站各车道通行状态检测的方法。

技术介绍

[0002]随着中国高速公路的逐年扩建，高速公路上的车辆也日益增多。而作为高速公路的出入口，收费站在高速公路体系中有着举足轻重的地位，高速收费站处的秩序问题一直困扰着司乘们。
[0003]目前针对收费站车道通行状态这一任务，普遍使用文字识别技术。对文字识别的方法主要使用不同程度的OCR技术，该技术虽然识别效果好，但大多用于文本识别，而且要融合目标检测、目标分类、方向分类器等多项技术，落地成本较高。为此，我们提出一种基于深度学习的收费站各车道通行状态检测的方法。

技术实现思路

[0004]本专利技术的目的在于提供一种基于深度学习的收费站各车道通行状态检测的方法，以解决上述
技术介绍
中提出的问题。
[0005]为了解决上述技术问题，本专利技术提供如下技术方案：一种基于深度学习的收费站各车道通行状态检测的方法，包括：
[0006]S1：构建数本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于深度学习的收费站各车道通行状态检测的方法，其特征在于：包括：S1：构建数据集D1，并按时间顺序分组，同时，构建车道通行状态数据库D2；S2：将D1中的图片序列进行统一处理后，输入至两层卷积模块进行下采样，获取对应的图像特征；S3：将S2中提取到的特征输入到CBAM模块，即卷积注意力模块(Convolutional Block Attention Module)，得到空间注意力特征和通道注意力特征的加权特征图，响应梯度变化较大的部分，加强特征提取；S4：将S3中得到的加权特征图输入到两个连续的密集连接模块中，此操作方便每组图片内的不同图片间的梯度互相融合，便于模糊图片的特征学习；S5：将S4中得到的密集连接学到到的特征图进行最大池化、全连接操作之后，将其结果输入softmax进行分类，对D1中每组收费站LED灯上文字图片使用AlexNet进行训练，得到分类模型；S6：将待测图片输入到分类模型中，可以得到LED灯上的文字；S7：根据S6得到的LED灯上文字及车道通行状态数据库D2判断该车道的通行状态。2.根据权利要求1所述的一种基于深度学习的收费站各车道通行状态检测的方法，其特征在于：所述CBAM模块为一种用于前馈卷积神经网络的注意力模块，其集成了通道注意力机制和空间注意力机制，使特征会沿着两个独立的维度依次判断特征图，进而得到两个维度上的特征权重。3.根据权利要求2所述的一种基于深度学习的收费站各车道通行状态检测的方法，其特征在于：所述通道注意力机制可以表达为公式(1)：其中：F表示输入特征图，M
c
∈R
C
×1×1表示通道注意力，σ表示sigmoid函数，MLP表示多层感知机，AvgPool表示全局平均池化，Ma...

【专利技术属性】
技术研发人员：申元，颜世航，
申请(专利权)人：杭州像素元科技有限公司，
类型：发明
国别省市：