[发明专利]一种基于车载移动深度学习平台的杆号识别方法

权利要求书

1.一种基于车载移动深度学习平台的杆号识别方法，其特征在于，包括以下步骤：

步骤1：获取高铁接触网零部件图像并标记，再筛选得到接触网杆号图像的数据集；获取的高铁接触网零部件图像为高铁检测车天窗拍摄收集的图像，并通过人工标记的方式生成含有零部件位置及种类信息的xml文件，其中数字1被标记为“1”，数字2被标记为“2”，以此类推；

步骤2：使用加速区域卷积神经网络模型Faster R-CNN对步骤1的接触网杆号图像数据集进行训练，对全局接触网杆号图像中的杆号牌定位，得到杆号牌图片数据集；

步骤3：使用快速定位模型SSD对步骤2的杆号牌图片数据集进行训练，对杆号牌图片中的数字定位，得到杆号牌数字图片数据集；

步骤4：在嵌入式深度学习移动平台部署深度学习模型；

步骤5：使用多类别非极大抑制算法对步骤4得到的识别目标进行筛选，得到杆号数字文本，具体为：

S51：通过新的图片输入到步骤4中的嵌入式深度学习平台，经过目标定位和分类，得到每个杆号数字定位框位置(x,y,w,h)、概率p及类别c，但仍会存在多个误识别框或一个数字被多个同类框定位；

S52：根据步骤S51得到的定位框坐标(x,y,w,h)、概率p及类别c，将其按照概率大小降序排列得到序列L；

S53：按照S52得到排列序列，从序列L中第一个定位框L(0)开始依次对其他定位框做交并比IOU计算，如果存在其他定位框L(i)与概率最大定位框L(0)的IOU大于0.5，则删除该定位框L(i)；

交并比IOU计算公式如下：

式中，A和B是定位框L(i)与概率最大定位框L(0)两个定位框区域；

S54：保留第一个定位框L(0)从降序序列L中去除该定位框，返回S53直至降序序列L为空；

S55：遍历保存的定位框，计算每个定位框的中心位置(x_m,y_m)，按照位置从上到下排列；

定位框中心位置计算公式如下：

式中，x_m和y_m是定位框中心点坐标，x、y、w和h分别为定位框的左上角横、纵坐标、定位框的宽和高；

S56：按照S55的排列顺序遍历保存的定位框，计算相邻两个定位框的中心距离，如果小于杆号牌像素大小的九分之一，则删除两者置信度低的定位框。

2.根据权利要求1所述的一种基于车载移动深度学习平台的杆号识别方法，其特征在于，所述步骤4中嵌入式深度学习移动平台部署过程如下：

S41：所使用的嵌入式移动深度学习平台为Nvidia Jetson TX2，该平台面向移动嵌入式系统市场中的GPU加速并行处理；

S42：对深度学习平台进行刷机并安装JetPack3.1与CTI-L4T，开启平台多核模式，编译安装tensorflow1.3.0版本。