[发明专利]网络训练方法、图像处理方法、装置、设备及存储介质

权利要求书

1.一种网络训练方法，其特征在于，包括：

将样本图像输入第i-1状态的图像处理网络，得到所述样本图像的多个第一处理结果，所述图像处理网络包括主干网络及多个分支网络，所述主干网络用于提取图像特征，所述多个分支网络用于基于所述图像特征输出各图像处理任务的处理结果，i为正整数；

根据所述样本图像的标注信息以及所述多个第一处理结果，分别确定各个分支网络的网络损失以及各个网络损失对于所述主干网络的梯度信息；

根据所述梯度信息，确定所述各个网络损失的目标加权系数；

根据所述各个网络损失及所述各个网络损失的目标加权系数，得到所述图像处理网络的总体损失；

根据所述总体损失，对所述第i-1状态的图像处理网络进行训练，得到第i状态的图像处理网络。

2.根据权利要求1所述的方法，其特征在于，所述根据所述梯度信息，确定所述各个网络损失的目标加权系数，包括：

根据所述梯度信息以及所述梯度信息的加权参数，确定总体梯度信息；

在所述总体梯度信息在各个梯度信息的方向上的梯度分量相同的情况下，将各个梯度信息的加权参数对应的参数值，确定为各个网络损失的目标加权系数。

3.根据权利要求1或2所述的方法，其特征在于，所述各个网络损失的目标加权系数之和为1。

4.根据权利要求1-3任一项所述的方法，其特征在于，根据所述总体损失，对所述第i-1状态的图像处理网络进行训练，包括：

根据所述总体损失，更新第i-1状态的主干网络的网络参数；

和/或，根据所述各个网络损失，对应更新第i-1状态的各个分支网络的网络参数。

5.根据权利要求4所述的方法，其特征在于，所述根据所述各个网络损失，对应更新第i-1状态的各个分支网络的网络参数，包括：

根据所述各个网络损失及所述各个网络损失的第i-1状态的缩放参数，确定与所述各个网络损失对应的缩放损失；

根据各个缩放损失，对应更新第i-1状态的各个分支网络的网络参数以及第i-1状态的缩放参数，以得到第i状态的各个分支网络及第i状态的缩放参数。

6.根据权利要求1-5任一项所述的方法，其特征在于，所述根据所述各个网络损失及所述各个网络损失的目标加权系数，得到所述图像处理网络的总体损失，包括：

根据所述各个网络损失的目标加权系数，对多个网络损失进行加权求和，得到所述总体损失。

7.根据权利要求1所述的方法，其特征在于，分别确定各个网络损失对于所述主干网络的梯度信息，包括：

分别确定所述各个网络损失对于所述主干网络提取的图像特征的梯度信息。

8.根据权利要求1-7任一项所述的方法，其特征在于，所述方法还包括：

在所述第i状态的图像处理网络满足训练条件的情况下，将所述第i状态的图像处理网络确定为训练后的图像处理网络。

9.根据权利要求1-8任一项所述的方法，其特征在于，所述图像处理任务包括图像识别、图像分类、图像分割、关键点检测中的至少两种。

10.一种图像处理方法，其特征在于，包括：

将待处理的图像输入至图像处理网络中，得到所述图像的多个第二处理结果，其中，所述图像处理网络是根据权利要求1-9任一项所述的网络训练方法训练得到的。