神经网络训练方法及装置以及图像处理方法及装置.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113743535A(43)申请公布日2021.12.03(21)申请号202111108379.4(22)申请日2019.05.21(62)分案原申请数据201910426010.42019.05.21(71)申请人北京市商汤科技开发有限公司地址100080北京市海淀区北四环西路58号11层1101-1117室(72)发明人韩江帆罗平王晓刚(74)专利代理机构北京林达刘知识产权代理事务所(普通合伙)11277代理人刘新宇(51)Int.Cl.G06K9/62(2006.01)G06N3/04(2006.01)G06N3/08(2006.01)权利要求书3页说明书14页附图3页(54)发明名称神经网络训练方法及装置以及图像处理方法及装置(57)摘要本公开涉及一种神经网络训练方法及装置以及图像处理方法及装置。该训练方法包括：通过神经网络对训练集中的目标图像进行分类处理，得到所述目标图像的预测分类结果；根据所述预测分类结果、所述目标图像的初始类别标签及校正类别标签，训练所述神经网络。本公开实施例可通过初始和校正类别标签共同监督神经网络的训练过程，简化训练过程和网络结构。CN113743535ACN113743535A权利要求书1/3页1.一种神经网络训练方法，其特征在于，包括：通过神经网络对训练集中的目标图像进行分类处理，得到所述目标图像的预测分类结果；根据所述预测分类结果、所述目标图像的初始类别标签及校正类别标签，训练所述神经网络，所述神经网络包括N个训练状态，N为大于1的整数；其中，所述根据所述预测分类结果、所述目标图像的初始类别标签及校正类别标签，训练所述神经网络，包括：根据第i状态的预测分类结果、所述目标图像的初始类别标签及第i状态的校正类别标签，确定所述神经网络的第i状态的总体损失，所述第i状态为所述N个训练状态中的一个，且0≤i<N；根据所述第i状态的总体损失，调整第i状态的神经网络的网络参数，得到第i+1状态的神经网络；其中，所述根据第i状态的预测分类结果、所述目标图像的初始类别标签及第i状态的校正类别标签，确定所述神经网络的第i状态的总体损失，包括：根据所述第i状态的预测分类结果以及所述目标图像的初始类别标签，确定所述神经网络的第i状态的第一损失；根据所述第i状态的预测分类结果以及所述目标图像的第i状态的校正类别标签，确定所述神经网络的第i状态的第二损失；根据所述第i状态的第一损失和所述第i状态的第二损失，确定所述神经网络的第i状态的总体损失。2.根据权利要求1所述的方法，其特征在于，所述神经网络包括特征提取网络和分类网络，其中，所述通过神经网络对训练集中的目标图像进行分类处理，得到所述目标图像的预测分类结果，包括：通过第i状态的特征提取网络对目标图像进行特征提取，得到所述目标图像的第i状态的第一特征；通过第i状态的分类网络对所述目标图像的第i状态的第一特征进行分类，得到所述目标图像的第i状态的预测分类结果。3.根据权利要求2所述的方法，其特征在于，所述方法还包括：通过第i状态的特征提取网络对训练集中第k个类别的多个样本图像进行特征提取，得到所述多个样本图像的第i状态的第二特征，所述第k个类别是所述训练集中的样本图像的K个类别中的一个，K为大于1的整数；对所述第k个类别的多个样本图像的第i状态的第二特征进行聚类处理，确定所述第k个类别的第i状态的类原型特征；根据K个类别的第i状态的类原型特征以及所述目标图像的第i状态的第一特征，确定所述目标图像的第i状态的校正类别标签。4.根据权利要求3所述的方法，其特征在于，所述根据K个类别的第i状态的类原型特征以及所述目标图像的第i状态的第一特征，确定所述目标图像的第i状态的校正类别标签，包括：2CN113743535A权利要求书2/3页分别获取所述目标图像的第i状态的第一特征与K个类别的第i状态的类原型特征之间的第一特征相似度；根据与第一特征相似度的最大值对应的类原型特征所属的类别，确定所述目标图像的第i状态的校正类别标签。5.根据权利要求4所述的方法，其特征在于，每个类别的第i状态的类原型特征包括多个类原型特征，其中，所述分别获取所述目标图像的第i状态的第一特征与K个类别的第i状态的类原型特征之间的第一特征相似度，包括：获取所述第i状态的第一特征与第k个类别的第i状态的多个类原型特征之间的第二特征相似度；根据所述第二特征相似度，确定所述第i状态的第一特征与第k个类别的第i状态的类原型特征之间的第一特征相似度。6.根据权利要求3‑5中任意一项所述的方法，其特征在于，所述第k个类别的第i状态的类原型特征包括所述第k个类别的多个样本图像的第i状态的第二特征的类中心。7.一种图像处理方法，其特征在于，所述