一种遮挡下的车辆识别方法及装置制造方法及图纸

技术编号:24093344 阅读:30 留言:0更新日期:2020-05-09 09:05
本发明专利技术公开了一种遮挡下的车辆识别方法及装置,所述方法包括:通过物体前景检测获取待识别图像中车辆区域的第一掩码图,通过所述第一掩码图获得车辆区域的第二掩码图;将所述第二掩码图与第一掩码图相减,得到非车辆空白图;获取所述多个第一区域的内顶点;以所述内顶点为端点构建的分割线划分所述第一掩码图中的连续区域为多个第二区域,根据所述多个第二区域的外顶点以及相邻内顶点构成的分割线构建多个矩形框;获取对应的单个车辆图像;将所述单个车辆图像输入训练好的识别神经网络进行识别,获得所述车辆的分类信息。本发明专利技术可以有效解决遮挡车辆识别问题,且模型计算量较小,可以用于遮挡车辆的实时识别。

A vehicle recognition method and device under occlusion

【技术实现步骤摘要】
一种遮挡下的车辆识别方法及装置
本专利技术涉及车辆识别领域,具体涉及一种遮挡下的车辆识别方法、装置、终端设备及计算机可读介质。
技术介绍
随着计算机视觉技术的发展,通过监控抓拍车辆图像,识别车辆类型已经成为交通智能管理的重要组成部分,得到了广泛的研究。但有时候抓拍的图像中车辆被其他车辆或其他物体遮挡,无法分割单一车辆图像,降低了车辆识别准确率。目前处理车辆遮挡的方法大多基于复杂模型,计算复杂度高,计算成本大,严重制约了方法的实际应用。
技术实现思路
有鉴于此,本专利技术的目的在于提供一种遮挡下的车辆识别方法、装置、终端设备及计算机可读介质,能够提高车辆识别准确率,解决了现有技术中遮挡下的车辆无法分割单一车辆图像,或分割计算复杂度高,计算成本大的问题。本专利技术实施例的第一方面提供了一种遮挡下的车辆识别方法,包括:通过物体前景检测获取待识别图像中车辆区域的第一掩码图,通过将所述第一掩码图中各连续区域的外边缘及外顶点连接,获得车辆区域的第二掩码图;将所述第二掩码图与第一掩码图相减,得到非车辆空白图,所述非本文档来自技高网...

【技术保护点】
1.一种遮挡下的车辆识别方法,其特征在于,包括:/n通过物体前景检测获取待识别图像中车辆区域的第一掩码图,通过将所述第一掩码图中各连续区域的外边缘及外顶点连接,获得车辆区域的第二掩码图;/n将所述第二掩码图与第一掩码图相减,得到非车辆空白图,所述非车辆空白图包括多个第一区域;/n获取所述多个第一区域的内顶点;/n以所述内顶点为端点构建的分割线划分所述第一掩码图中的连续区域为多个第二区域,根据所述多个第二区域的外顶点以及相邻内顶点构成的分割线构建多个矩形框;/n获取待识别图像中所述多个矩形框位置分别对应的单个车辆图像;/n将所述单个车辆图像输入训练好的识别神经网络进行识别,获得所述车辆的分类信息...

【技术特征摘要】
1.一种遮挡下的车辆识别方法,其特征在于,包括:
通过物体前景检测获取待识别图像中车辆区域的第一掩码图,通过将所述第一掩码图中各连续区域的外边缘及外顶点连接,获得车辆区域的第二掩码图;
将所述第二掩码图与第一掩码图相减,得到非车辆空白图,所述非车辆空白图包括多个第一区域;
获取所述多个第一区域的内顶点;
以所述内顶点为端点构建的分割线划分所述第一掩码图中的连续区域为多个第二区域,根据所述多个第二区域的外顶点以及相邻内顶点构成的分割线构建多个矩形框;
获取待识别图像中所述多个矩形框位置分别对应的单个车辆图像;
将所述单个车辆图像输入训练好的识别神经网络进行识别,获得所述车辆的分类信息。


2.根据权利要求1所述的遮挡下的车辆识别方法,其特征在于,通过物体前景检测获取待识别图像中车辆区域的第一掩码图,包括:
获取待识别图像的视频流,采用Codebook算法从所述视频流中提取前景图像;
对所述前景图像做形态操作,所述形态操作包括膨胀和侵蚀;
对形态操作后的所述前景图像采用Canny算法获取轮廓,所述轮廓内区域即为车辆区域的第一掩码图。


3.根据权利要求2所述的遮挡下的车辆识别方法,其特征在于,所述采用Codebook算法从所述视频流中提取前景图像,包括:
从所述视频流中提取待识别图像,计算待识别图像的全局平均灰度值;
根据所述全局平均灰度值查找所对应的编码本,若查找成功,则待识别图像的所有像素均为前景,若查找失败,遍历待识别图像的每个像素的每个编码元,如果存在匹配当前像素值的编码元,则当前像素为背景,否则,当前像素为前景。


4.根据权利要求2所述的遮挡下的车辆识别方法,其特征在于,所述对形态操作后的所述前景图像采用Canny算法获取轮廓,包括:
使用高斯滤波平滑所述前景图像;
应用高斯模糊去除所述前景图像的噪声,降低伪边缘的识别;
计算所述前景图像的梯度幅值和梯度方向,其中,梯度模G的计算公式如下:



式中:Gx为水平方向的差分,Gy为垂直方向的差分;
梯度方向θ的计算公式如下:
θ=atan2(Gy,Gx);
遍历所述前景图像中轮廓曲线模糊部分的像素点,比较当前像素点的梯度强度和正负梯度方向的梯度强度,如果当前像素点的梯度强度大于同方向的其他点的梯度强度,则保留其值,否则,将该值设为0,将保留的所有像素点拼接成轮廓。


5.根据权利要求1所述的遮挡下的车辆识别方法,其特征在于,
所述识别神经网络包括卷积模...

【专利技术属性】
技术研发人员:罗茜张斯尧王思远谢喜林黄晋文戎张诚
申请(专利权)人:湖南千视通信息科技有限公司
类型:发明
国别省市:湖南;43

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1