[发明专利]网络训练及设备的控制方法、装置、设备及存储介质

说明书

本公开提供了一种网络训练方法、设备的控制方法、装置、电子设备及存储介质，该方法包括：确定待训练的目标网络以及待训练的目标网络的期望性能信息；基于待训练的目标网络的期望性能信息，从训练数据库中确定多张携带有标注信息的训练图像样本，构成训练图像样本集合；基于训练图像样本集合对待训练的目标网络进行训练，并对训练后的中间网络进行测试，根据测试结果对训练图像样本集合进行调整，重复训练、测试和对所述训练图像样本集合进行调整的步骤，直至符合预设条件。本公开实施例，根据待训练网络的期望性能自动确定训练图像样本集合，并且可以基于测试结果自动调整训练图像样本集合，有利于节约人力成本以及提升目标网络的训练效率。

技术领域

本公开涉及计算机视觉技术领域，具体而言，涉及一种网络训练方法、智能行驶设备的控制方法、装置、电子设备以及计算机可读存储介质。

背景技术

目前，网络模型已经在多个领域(例如，自动驾驶领域)中得到应用，在网络模型的构建过程中，经常会对待训练的网络模型进行训练，进而利用训练好的网络模型完成工作任务。

在现有的网络模型的训练过程中，为了提高网络模型的精度，通常会采用手动切换数据集或向数据集中添加数据的方法，然而，此种方法不仅影响网络模型的开发进度，还占用大量的人力和时间成本。

发明内容

本公开实施例提供了一种网络训练方法，包括：

确定待训练的目标网络以及所述待训练的目标网络的期望性能信息；

基于所述待训练的目标网络的期望性能信息，从训练数据库中确定多张携带有标注信息的训练图像样本，构成训练图像样本集合；所述期望性能信息与所述训练图像样本集合中的不同标注类别的训练图像样本数量相关；

基于所述训练图像样本集合对所述待训练的目标网络进行训练，并对训练后的中间网络进行测试，根据测试结果对所述训练图像样本集合进行调整，重复训练、测试和对所述训练图像样本集合进行调整的步骤，直至符合预设条件。

本公开实施例中，基于待训练的目标网络的期望性能信息，可以从训练数据库中自动确定与待训练的目标网络匹配的训练图像样本集合，并且可以根据测试结果自动调整训练图像样本集合，使得调整后的训练图像样本集合更加符合训练要求，如此，不仅可以节省人力和时间成本，还可以提高目标网络的训练效率。

一种可选的实施方式中，基于所述训练图像样本集合对所述待训练的目标网络进行训练，并对训练后的中间网络进行测试，根据测试结果对所述训练图像样本集合进行调整，重复训练、测试和对所述训练图像样本集合进行调整的步骤，直至符合预设条件，包括：

基于所述训练图像样本集合对所述待训练的目标网络进行训练，得到中间网络；

对所述中间网络进行测试，得到测试结果；

基于所述测试结果，从所述训练数据库中重新确定所述训练图像样本集合；

利用重新确定的训练图像样本集合对测试后的中间网络进行训练，并对训练后的中间网络进行测试，以及基于测试结果再次从所述训练数据库中重新确定所述训练图像样本集合，直至符合预设条件，得到训练好的目标网络。

本公开实施例中，根据测试结果重新确定训练图像样本集合，并基于重新确定的训练图像样本集合重复对中间网络进行训练，如此，可以提高目标网络的训练精度以及训练效率。

一种可选的实施方式中，所述得到训练好的目标网络，包括：

基于各个中间网络对应的测试结果，确定所述训练好的目标网络。

本公开实施例中，由于每个阶段的中间网络的检测性能均不同，对应的测试结果也会存在不同，通过该测试结果确定训练好的目标网络，如此，可以确定出最优且最适配的目标网络。