目标识别网络的训练方法、装置和系统.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN114998687A(43)申请公布日2022.09.02(21)申请号202210681687.4(22)申请日2022.06.15(71)申请人上海商汤智能科技有限公司地址200233上海市徐汇区桂平路391号3号楼1605A室(72)发明人金国强孙明珊李韡暴天鹏吴立威(74)专利代理机构北京博思佳知识产权代理有限公司11415专利代理师董晓盈(51)Int.Cl.G06V10/774(2022.01)G06V10/764(2022.01)G06V10/82(2022.01)权利要求书2页说明书11页附图2页(54)发明名称目标识别网络的训练方法、装置和系统(57)摘要公开了一种目标识别网络的训练方法、装置和系统。所述训练方法，包括：获取接触网图像，并获取所述接触网图像中至少一个对象的预选框；利用目标识别网络中的第一目标识别分支网络对所述接触网图像进行处理，得到第一识别结果，并利用所述目标识别网络中的第二目标识别分支网络对所述接触网图像进行处理，得到第二识别结果；根据所述预选框与所述第一识别结果之间的第一差异，以及所述第一识别结果和所述第二识别结果之间的第二差异，调整所述第一目标识别分支网络的网络参数。CN114998687ACN114998687A权利要求书1/2页1.一种目标识别网络的训练方法，其特征在于，包括：获取接触网图像中至少一个对象的预选框；利用目标识别网络中的第一目标识别分支网络对所述接触网图像进行处理，得到第一识别结果，并利用所述目标识别网络中的第二目标识别分支网络对所述接触网图像进行处理，得到第二识别结果；根据所述预选框与所述第一识别结果之间的第一差异，以及所述第一识别结果和所述第二识别结果之间的第二差异，调整所述第一目标识别分支网络的网络参数。2.根据权利要求1所述的方法，其特征在于，所述利用目标识别网络中的第一目标识别分支网络对所述接触网图像进行处理，得到第一识别结果，并利用所述目标识别网络中的第二目标识别分支网络对所述接触网图像进行处理，得到第二识别结果，包括：对所述接触网图像采用第一增广方式增广，得到第一增广结果；将所述第一增广结果输入目标识别网络中的第一目标识别分支网络，得到第一识别结果，所述第一识别结果包括至少一个对象的第一检测框；对所述接触网图像采用第二增广方式增广，得到第二增广结果，所述第二增广方式不同于所述第一增广方式；将所述第二增广结果输入所述目标识别网络中的第二目标识别分支网络，得到第二识别结果，所述第二识别结果包括至少一个对象的第二检测框。3.根据权利要求1所述的方法，其特征在于，所述调整所述第一目标识别分支网络的网络参数，包括：在每一轮迭代中，采用梯度下降法调整所述第一目标识别分支网络的网络参数，得到调整后的第一目标识别分支网络的网络参数；所述调整所述第一目标识别分支网络的网络参数之后，还包括：基于所述调整后的第一目标识别分支网络的网络参数，以及上一轮第二目标识别分支网络的网络参数，确定所述第二目标识别分支网络的网络参数。4.根据权利要求1‑3中任意一项所述的方法，其特征在于，所述方法还包括：获取包括目标对象的标注信息的接触网图像作为样本图像，其中，所述标注信息包括标注框以及所述标注框的标注类别信息，所述标注类别信息指示所述标注框内目标对象的类别；利用所述第一目标识别分支网络对所述样本图像进行识别，得到所述样本图像的识别结果，所述识别结果包括第三检测框以及所述第三检测框对应图像区域的预测类别信息；根据所述标注框与所述第三检测框之间的差异，以及所述标注类别信息与所述预测类别信息的差异，调整所述第一目标识别分支网络的网络参数。5.根据权利要求4所述的方法，其特征在于，所述利用第一目标识别分支网络对所述样本图像进行识别之前，还包括：根据目标对象的类别数量，调整所述第一目标识别分支网络的输出层网络参数。6.根据权利要求4所述的方法，其特征在于，所述目标对象至少包括下列中的至少一个：绝缘子、驱鸟罩、接触线、吊弦和承力索。7.根据权利要求1所述的方法，其特征在于，所述获取接触网图像中至少一个对象的预选框，包括：2CN114998687A权利要求书2/2页获取接触网图像中的多个图像区域；根据各图像区域间的相似度，合并各图像区域；根据合并后的图像区域，确定至少一个对象的预选框。8.一种目标检测方法，其特征在于，所述方法包括：获取待检测图像；将所述待检测图像输入第一目标识别分支网络，输出识别结果，所述第一目标识别分支网络基于权利要求1‑7中任意一项权利要求所述方法得到。9.一种目标识别网络的训练装置，其特征在于，包括：第一获取单元，用于获取接触网图像中至少一个对象的预选框；处理单元，用于利用目标识别网络中的第一目标识别分支网络对所述接