[发明专利]目标识别网络的训练方法、装置和系统

说明书

公开了一种目标识别网络的训练方法、装置和系统。所述训练方法，包括：获取接触网图像，并获取所述接触网图像中至少一个对象的预选框；利用目标识别网络中的第一目标识别分支网络对所述接触网图像进行处理，得到第一识别结果，并利用所述目标识别网络中的第二目标识别分支网络对所述接触网图像进行处理，得到第二识别结果；根据所述预选框与所述第一识别结果之间的第一差异，以及所述第一识别结果和所述第二识别结果之间的第二差异，调整所述第一目标识别分支网络的网络参数。

技术领域

本说明书涉及计算机技术领域，尤其涉及一种目标识别网络的训练方法、装置及系统。

背景技术

随着公共交通建设的不断发展，对部署在公共交通道路上供电系统的安全性与可靠性的要求日渐提升。以公共交通中的高铁为例，接触网是高铁安全运行的关键组成部分。

应用计算机技术辅助的高铁接触网检测是铁路轨道维护的发展趋势，然而目前的检测方法，需要人工对海量接触网图像进行标注以训练神经网络，存在人工标注成本高的缺陷。

发明内容

本公开提供了一种目标识别网络的训练方案。

根据本公开的一方面，提供一种目标识别网络的训练方法，包括：

获取接触网图像中至少一个对象的预选框；利用目标识别网络中的第一目标识别分支网络对所述接触网图像进行处理，得到第一识别结果，并利用所述目标识别网络中的第二目标识别分支网络对所述接触网图像进行处理，得到第二识别结果；根据所述预选框与所述第一识别结果之间的第一差异，以及所述第一识别结果和所述第二识别结果之间的第二差异，调整所述第一目标识别分支网络的网络参数。

结合本公开提供的任一实施方式，所述利用目标识别网络中的第一目标识别分支网络对所述接触网图像进行处理，得到第一识别结果，并利用所述目标识别网络中的第二目标识别分支网络对所述接触网图像进行处理，得到第二识别结果，包括：对所述接触网图像采用第一增广方式增广，得到第一增广结果；将所述第一增广结果输入目标识别网络中的第一目标识别分支网络，得到第一识别结果，所述第一识别结果包括至少一个对象的第一检测框；对所述接触网图像采用第二增广方式增广，得到第二增广结果，所述第二增广方式不同于所述第一增广方式；将所述第二增广结果输入所述目标识别网络中的第二目标识别分支网络，得到第二识别结果，所述第二识别结果包括至少一个对象的第二检测框。

结合本公开提供的任一实施方式，所述调整所述第一目标识别分支网络的网络参数，包括：在每一轮迭代中，采用梯度下降法调整所述第一目标识别分支网络的网络参数，得到调整后的第一目标识别分支网络的网络参数；所述调整所述第一目标识别分支网络的网络参数之后，还包括：基于所述调整后的第一目标识别分支网络的网络参数，以及上一轮第二目标识别分支网络的网络参数，确定所述第二目标识别分支网络的网络参数。

结合本公开提供的任一实施方式，所述方法还包括：获取包括目标对象的标注信息的接触网图像作为样本图像，其中，所述标注信息包括标注框以及所述标注框的标注类别信息，所述标注类别信息指示所述标注框内目标对象的类别；利用所述第一目标识别分支网络对所述样本图像进行识别，得到所述样本图像的识别结果，所述识别结果包括第三检测框以及所述第三检测框对应图像区域的预测类别信息；根据所述标注框与所述第三检测框之间的差异，以及所述标注类别信息与所述预测类别信息的差异，调整所述第一目标识别分支网络的网络参数。

结合本公开提供的任一实施方式，所述利用第一目标识别分支网络对所述样本图像进行识别之前，还包括：根据目标对象的类别数量，调整所述第一目标识别分支网络的输出层网络参数。

结合本公开提供的任一实施方式，所述目标对象至少包括下列中的至少一个：绝缘子、驱鸟罩、接触线、吊弦和承力索。