一种识别模型的训练方法及装置制造方法及图纸

本申请实施例提供一种识别模型的训练方法及装置，用于提高图像的标注效率，进而优化识别模型。该方法具体可以包括：获取第一图像，其中第一图像可以包括第一语义对象和第二语义对象；根据第一语义对象和第二语义对象之间的位置关系，可以自动对第一语义对象和第二语义对象之间的边界信息进行自动标注，更新第一语义对象的边界框。如此，能够有效提升图像标注的效率，进而优化识别模型的精度和有效性。进而优化识别模型的精度和有效性。进而优化识别模型的精度和有效性。

【技术实现步骤摘要】
一种识别模型的训练方法及装置


[0001]本申请涉及图像处理
，尤其涉及一种识别模型的训练方法及装置。

技术介绍

[0002]在自动驾驶、智慧医疗与智慧城市等领域中，需要使用基于深度学习的识别模型来对应用场景中的目标物体进行识别，其中，该识别模型需要对大量已标注样本数据进行训练才能得到。
[0003]然而，该识别模型对应的样本数据多为图像，目前的识别模型无法精确识别出图像中不同语义对象之间的边界(例如，车辆和行人之间的边界)、图像中特定语义对象(例如，路面)所在的区域边界以及图像中被特定语义对象(例如，绿化带)隔离所形成的区域边界，需要人工大量标注，并且不同语义对象之间边界还存在重复标注的情况，导致此类样本数据存在标注效率低下的问题。
[0004]因此，如何提高标注数据的效率，获得准确率高的识别模型，是本申请所要解决的技术问题。

技术实现思路

[0005]本申请提供一种识别模型的训练方法及装置，用于提高图像的标注效率，进而优化识别模型。
[0006]第一方面，本申请实施例提供一种识别模型的训练方法，本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种识别模型的训练方法，其特征在于，包括：获取第一图像，所述第一图像包括至少两个语义对象，所述至少两个语义对象包括第一语义对象和第二语义对象；根据所述第一语义对象的第一边界框与所述第二语义对象的第二边界框的第一位置关系，更新所述第一语义对象的边界框，得到所述第一语义对象的更新后的第一边界框；其中，所述第一位置关系包括外切、内切、相交中的一种或多种；所述更新后的第一边界框与所述第二边界框具有相同的第一目标有向线段；基于所述第一图像、所述第二边界框、所述更新后的第一边界框，生成第二图像；将所述第二图像输入所述识别模型进行训练，得到训练后的识别模型。2.如权利要求1所述的方法，其特征在于，在更新所述第一语义对象的边界框之前，还包括：接收第一操作；根据所述第一操作，确定所述第一边界框为第一图层，所述第二边界框为第二图层，所述第一目标有向线段位于所述第二图层上。3.如权利要求1或2所述的方法，其特征在于，所述获取第一图像，包括：获得至少一个语义对象的边界框，将所述至少一个语义对象的边界框进行随机组合，得到所述第一图像；和/或，获取第三图像，对所述第三图像进行处理，得到所述第一图像。4.如权利要求1
‑
3任一项所述的方法，其特征在于，所述根据所述第一语义对象的第一边界框与所述第二语义对象的第二边界框的第一位置关系，更新所述第一语义对象的边界框，包括：确定所述第一边界框和所述第二边界框的至少两个交点；根据所述至少两个交点，将所述第一边界框拆分成至少两条第一有向线段，以及将所述第二边界框拆分成至少两条第二有向线段；每条所述第一有向线段和每条所述第二有向线段均包括起点和终点；根据所述第一位置关系，确定位于所述第二边界框以外的第一有向线段和位于所述第一边界框以内的第二有向线段；基于所述位于所述第二边界框以外的第一有向线段和所述位于所述第一边界框以内的第二有向线段，确定所述更新后的第一边界框。5.如权利要求4所述的方法，其特征在于，基于所述位于所述第二边界框以外的第一有向线段和所述位于所述第一边界框以内的第二有向线段，确定所述更新后的第一边界框，包括：对所述位于所述第二边界框以外的第一有向线段和所述位于所述第一边界框以内的第二有向线段进行连接，得到所述更新后的第一边界框。6.如权利要求4或5所述的方法，其特征在于，所述根据所述第一位置关系，确定所述位于所述第一边界框以内的第二有向线段，包括：当根据所述第一位置关系确定所述至少两条第二有向线段中存在位于所述第一边界框上的第二有向线段时，则确定该第二有向线段是位于所述第一边界框以内的第二有向线段。7.如权利要求4
‑
6任一项所述的方法，其特征在于，所述确定所述第一边界框和所述第
二边界框的至少两个交点，包括：确定任一所述第一有向线段与任一所述第二有向线段不平行，且存在第一交点；确定所述第一交点在任一所述第一有向线段和任一所述第二有向线段上；若所述第一交点不为任一所述第一有向线段的终点，以及不为任一所述第二有向线段的终点，将所述第一交点确定为所述至少两个交点中的任一交点。8.如权利要求7所述的方法，其特征在于，所述确定所述第一交点在任一第一有向线段和所述任一第二有向线段上，包括：确定所述第一交点与任一所述第一有向线段的距离小于第一阈值，确定所述第一交点在任一所述第一有向线段上；和/或，确定所述第一交点与任一所述第二有向线段的距离小于第二阈值，确定所述第一交点在任一所述第二有向线段上。9.如权利要求7或8所述的方法，其特征在于，所述确定所述第一边界框和所述第二边界框的至少两个交点，包括：确定任一所述第一有向线段与任一所述第二有向线段存在第二交点；当所述第二交点与所述第一交点的距离小于第三阈值时，将所述第二交点和所述第一交点作为同一个交点。10.如权利要求4
‑
6任一项所述的方法，其特征在于，所述确定所述第一边界框和所述第二边界框的至少两个交点，包括：确定任一所述第一有向线段与任一所述第二有向线段平行；任一所述第一有向线段包括第一起点和第一终点，任一所述第二有向线段包括第二起点和第二终点；确定所述第一起点或所述第一终点与任一所述第二有向线段的距离小于第四阈值，则确定任一所述第一有向线段与任一所述第二有向线段重合；或者，确定所述第二起点或所述第二终点与任一所述第一有向线段的距离小于第五阈值，则确定任一所述第一有向线段与任一所述第二有向线段重合；将所述第一起点和/或所述第二起点确定为所述至少两个交点中的任一交点。11.如权利要求10所述的方法，其特征在于，所述将所述第一起点和/或所述第二起点确定为所述至少两个交点中的任一交点，包括：当所述第二起点在任一所述第一有向线段上时，将所述第二起点确定为所述任一交点；或者，当所述第一起点在任一所述第二有向线段上时，将所述第一起点确定为所述任一交点。12.如权利要求1
‑
11任一项所述的方法，其特征在于，所述至少两个语义对象还包括第三语义对象；在得到所述第一语义对象的更新后的第一边界框之后，所述方法还包括：确定所述第三语义对象的第三边界框位于所述第一边界框内...

【专利技术属性】
技术研发人员：王庭杨，苏飞，
申请(专利权)人：华为技术有限公司，
类型：发明
国别省市：