[发明专利]一种图像处理方法、装置、电子设备及其存储介质

摘要

本发明实施例提供一种图像处理方法、装置、电子设备及其存储介质，涉及图像识别技术领域。该方法包括获取训练数据集并构建标签矩阵；利用训练数据集对深度神经网络中的主网络和多个子网络进行训练，获得第一输出矩阵和第二输出矩阵；根据各子网络对应的场景类别获得第一输出矩阵对应的第一切片矩阵和标签矩阵对应的第二切片矩阵；根据第二输出矩阵、第一切片矩阵和第二切片矩阵计算损失函数，并根据损失函数对所述深度神经网络中的参数进行优化，获得所述主网络对应的场景识别模型。装置用于执行上述方法。本发明实施例通过子网络来对主网络进行训练，从而获得到的场景识别模型在对图像进行场景识别时，能够同时保证高准确率和高召回率。

主权项

1.一种图像处理方法，其特征在于，包括：获取训练数据集，并根据所述训练数据集构建标签矩阵；利用所述训练数据集对深度神经网络中的主网络和多个子网络进行训练，获得所述主网络对应的第一输出矩阵和各所述子网络分别对应的第二输出矩阵；其中，所述主网络用于对所有场景类别进行识别，每一所述子网络用于对相应的场景类别进行识别；根据各所述子网络对应的场景类别分别对所述第一输出矩阵和所述标签矩阵进行切片划分，获得所述第一输出矩阵对应的第一切片矩阵和所述标签矩阵对应的第二切片矩阵；根据所述第二输出矩阵、所述第一切片矩阵和所述第二切片矩阵计算损失函数，并根据所述损失函数对所述深度神经网络中的参数进行优化，获得所述主网络对应的场景识别模型。