图像识别、图像识别模型的训练方法和装置制造方法及图纸

本公开提供了图像识别、图像识别模型的训练方法和装置，涉及人工智能技术领域，具体涉及计算机视觉和深度学习技术领域，可用于智慧城市和智能交通场景下。具体实施方式包括：将标注图像输入第一图像识别模型，将真实图像输入第二图像识别模型，其中，第一图像识别模型和第二图像识别模型为待训练的相同图像识别模型；基于第一图像识别模型中目标处理层的结果和第二图像识别模型中目标处理层的结果，生成第一图像识别模型的损失值；基于损失值，训练第一图像识别模型，得到训练后的第一图像识别模型。本公开可以提高训练的准确度。本公开可以提高训练的准确度。本公开可以提高训练的准确度。

【技术实现步骤摘要】
图像识别、图像识别模型的训练方法和装置


[0001]本公开涉及人工智能
，具体涉及计算机视觉和深度学习
，可用于智慧城市和智能交通场景下，尤其涉及图像识别模型的训练方法和装置。

技术介绍

[0002]图像识别，是指利用计算机对图像进行处理、分析和理解，以识别各种不同的目标和对象的技术，是应用深度学习算法的一种实践应用。举例来说，图像识别包括人脸识别与商品识别。
[0003]在相关技术中，可以采用图像识别来识别图像中的交通标线。该方式可以取代消耗人力的人工识别，从而减小识别误差，提高识别的稳定性和准确度。

技术实现思路

[0004]提供了一种图像识别模型的训练方法、装置、电子设备以及存储介质。
[0005]根据第一方面，提供了一种图像识别模型的训练方法，包括：将标注图像输入第一图像识别模型，将真实图像输入第二图像识别模型，其中，第一图像识别模型和第二图像识别模型为待训练的相同图像识别模型；基于第一图像识别模型中目标处理层的结果和第二图像识别模型中目标处理层的结果，生成第一图像识别模型的损失值；基于损失值本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像识别模型的训练方法，所述方法包括：将标注图像输入第一图像识别模型，将真实图像输入第二图像识别模型，其中，所述第一图像识别模型和所述第二图像识别模型为待训练的相同图像识别模型；基于所述第一图像识别模型中目标处理层的结果和所述第二图像识别模型中所述目标处理层的结果，生成所述第一图像识别模型的损失值；基于所述损失值，训练所述第一图像识别模型，得到训练后的第一图像识别模型。2.根据权利要求1所述的方法，其中，所述基于所述第一图像识别模型中目标处理层的结果和所述第二图像识别模型中所述目标处理层的结果，生成所述第一图像识别模型的损失值，包括：将所述第一图像识别模型中所述目标处理层的结果和所述第二图像识别模型中所述目标处理层的结果，输入预设的跨域损失函数，得到第一损失值；基于所述第一损失值，生成所述第一图像识别模型的损失值，其中，所述跨域损失函数是基于最大均值差异生成的。3.根据权利要求1所述的方法，其中，所述训练包括多次训练过程；每次训练过程包括以下步骤：按照前一次训练过程得到的第一图像识别模型的权重，更新前一次训练过程所采用的第二图像识别模型的权重，得到本次训练过程将采用的第二图像识别模型。4.根据权利要求1所述的方法，其中，所述目标处理层是中间处理层，在所述第一图像识别模型和所述第二图像识别模型两者的训练结构中包括所述目标处理层，所述目标处理层为全连接层。5.根据权利要求2所述的方法，其中，所述基于所述第一损失值，生成所述第一图像识别模型的损失值，包括：根据交叉熵损失函数和所述第一图像识别模型的输出，生成第二损失值；根据洛瓦斯损失函数和所述第一图像识别模型的输出，生成第三损失值；根据所述第一损失值、所述第二损失值和所述第三损失值，生成所述第一图像识别模型的损失值。6.根据权利要求1
‑
5之一所述的方法，其中，图像识别模型的输出包括掩膜，图像识别模型用于识别图像中的目标对象，所述掩膜用于指示输入图像识别模型的图像中的所述目标对象的类别和位置。7.根据权利要求6所述的方法，其中，所述标注图像的标注信息包括所述掩膜，所述掩膜包括图像的各个像素的掩膜，像素的掩膜包括预设的颜色信息，不同的颜色信息指示不同类别的交通标线。8.根据权利要求1所述的方法，其中，图像识别模型包括编码器和解码器；所述训练中的前向传播过程，包括：通过所述编码器，获取输入所述图像识别模型的图像的特征图，并对所述特征图进行金字塔池化；根据金字塔池化结果，生成编码器的特征编码结果；通过所述解码器，对所述特征编码结果和所述特征图进行特征融合；根据所述特征融合结果，得到所输入图像的掩膜；
其中，图像识别模型包括目标卷积层，所述目标卷积层用于进行深度可分离卷积处理和膨胀卷积处理。9.一种图像识别方法，其中，所述方法采用权利要求1
‑
8中任一项中训练后的图像识别模型，图像识别模型的输出包括掩膜，图像识别模型用于识别图像中的目标对象，所述目标对象为交通标线。10.根据权利要求9所述的方法，其中，所述方法还包括：获取输出的掩膜指示的目标交通标线的定位信息；在交通标线集合中，确定所述定位信息对应的交通标线参考图；根据所述交通标线参考图，确定所述目标交通标线的缺失情况信息，其中，所述缺失情况信息指示交通标线是否存在缺失。11.根据权利要求9所述的方法，其中，所述根据所述交通标线参考图，确定所述目标交通标线的缺失情况信息，包括：确定所述目标交通标线与所述交通标线参考图之间的面积比值；根据所述面积比值，确定所述目标交通标线是否存在缺失。12.一种图像识别模型的训练装置，所述装置包括：输入单元，被配置成将标注图像输入第一图像识别模型，将真实图像输入第二图像识别模型，其中，所述第一图像识别模型和所述第二图像识别模型为待训练的相同图像识别模型；生成单元，被配置成基于所述第一图像识别模型中目标处理层的结果和所述第二图像识别模型中所述目标处理层的结果，生成所述第一图像识别模型的损失值；训练单元，被配置成基于所述损失值，训练所述第一图像识别模型，得到训练后的第一图像识别模型。13.根据权利要求12所述的装置，其中，所述生成单元，进一步被配置成...

【专利技术属性】
技术研发人员：刘旭辉，谷祎，刘静伟，王晓迪，韩树民，冯原，辛颖，李超，龙翔，彭岩，郑弘晖，张滨，王云浩，
申请(专利权)人：北京百度网讯科技有限公司，
类型：发明
国别省市：