[发明专利]一种图像的识别方法和装置

权利要求书

1.一种图像的识别方法，其特征在于，包括：

获取轨道交通车辆的待测图像；

基于目标信息识别模型对所述待测图像进行目标识别，以得到所述待测图像中包含的目标对象的目标信息，其中，所述目标信息识别模型为使用训练图像对初始信息识别模型进行训练得到的，所述训练图像为对初始图像进行数据增强处理后得到的，所述初始图像为标注了目标区域的标签信息的图像。

2.根据权利要求1所述的方法，其特征在于，基于所述目标信息识别模型大于所述待测图像进行所述目标识别之前，所述方法还包括：

在图像训练集中的包含的多个初始图像中获取待处理的第一图像，以及预定数量的第二图像，其中，所述第一图像与所述第二图像属性信息不同；

对所述第一图像以及所述第二图像进行所述数据增强处理，以得到所述训练图像；

基于所述训练图像对所述初始信息识别模型进行训练，以得到所述目标信息识别模型。

3.根据权利要求2所述的方法，其特征在于，对所述第一图像以及所述第二图像进行所述数据增强处理，以得到所述训练图像包括：

对所述第一图像以及所述第二图像进行亮度调节，以得到调节后的第三图像；

对所述第三图像进行背景替换，以得到第四图像；

对所述第四图像按照目标位置关系进行图像拼接，以得到第五图像；

将所述第五图像确定为所述训练图像。

4.根据权利要求3所述的方法，其特征在于，对所述第一图像以及所述第二图像进行亮度调节，以得到调节后的所述第三图像包括：

获取所述图像训练集的参考亮度值；

确定所述第一图像以及所述第二图像的初始图像亮度值；

比较所述初始图像亮度值以及所述参考亮度值的大小；

在所述初始图像亮度值小于所述参考亮度值的情况下，将对应图像的所述初始图像亮度值向上调整至第一亮度值，以得到所述第三图像，其中，所述第一亮度值在设定阈值范围内；

在所述初始图像亮度值大于所述参考亮度值的情况下，将对应图像的所述初始图像亮度值向下调整至第二亮度值，以得到所述第三图像，其中，所述第二亮度值在所述设定阈值范围内。

5.根据权利要求4所述的方法，其特征在于，获取所述图像训练集的所述参考亮度值包括：

确定所述图像训练集中包含的每个初始图像的初始亮度比值，其中所述初始亮度比值为每个所述初始图像的亮度与所述初始图像中的像素数量的比值；

在多个所述初始亮度比值中筛选出大于设定阈值的目标亮度比值；

确定所述目标亮度比值的平均值为所述图像训练集的所述参考亮度值。

6.根据权利要求3所述的方法，其特征在于，对所述第三图像进行背景替换，以得到所述第四图像包括：

获取所述第三图像中的所述目标区域所述标签信息；

在所述标签信息为第一标签信息时，基于预设概率对所述目标区域进行所述背景替换，以得到第六图像，其中，所述背景替换为将背景图中任意区域的图像替换为所述目标区域的图像，所述背景图为从所述图像训练集中挑选出的图像，所述背景图与所述第一图像的属性信息不同；

对所述第六图像进行目标处理，以得到所述第四图像，其中，所述目标处理包括翻转、旋转、加噪声、镜像。

7.根据权利要求6所述的方法，其特征在于，在获取所述第三图像中的所述目标区域所述标签信息之后，所述方法还包括：

在所述标签信息为第二标签信息时，对所述第三图像进行所述目标处理，以得到所述第四图像。

8.一种图像的识别装置，其特征在于，包括：

第一获取模块，用于获取轨道交通车辆的待测图像；

识别模块，用于基于目标信息识别模型对所述待测图像进行目标识别，以得到所述待测图像中包含的目标对象的目标信息，其中，所述目标信息识别模型为使用训练图像对初始信息识别模型进行训练得到的，所述训练图像为对初始图像进行数据增强处理后得到的，所述初始图像为标注了目标区域的标签信息的图像。