有效

类相关性预测改进方法、装置、计算机设备及存储介质

杜杰、王晶、刘鹏、汪天富
深圳大学

摘要

本发明实施例公开了类相关性预测改进方法、装置、计算机设备及存储介质。所述方法包括:获取待预测图像,以得到样本集;构建预测网络以及损失函数;利用所述样本集对所述预测网络进行训练,对训练所得的结果进行规范化处理,以得到预测概率值;采用所述损失函数结合所述预测概率值对训练后的预测网络进行调整;其中,所述损失函数包括交叉熵损失函数以及类损失函数;所述类损失函数包括用于拉近相似类的网络输出值产生的损失的类内损失函数以及用于拉远不相似类的预测产生的损失的类间损失函数。通过实施本发明实施例的方法可实现在保证准确率的情况下,利用先验知识来拉近相似的两类的网络输出值,拉远不相似的两类的网络输出值,提高神经网络的泛化能力的目的。