[发明专利]图像预测方法、装置、计算机设备和存储介质

说明书

本申请涉及一种图像预测方法、装置、计算机设备和存储介质。所述方法包括：获取待测图像；待测图像中包括多个图像区域，且不同的图像区域对应不同的分类类别；将待测图像输入预设的图像预测模型，得到预测图像；其中，图像预测模型为根据预设的损失函数训练得到的，预设的损失函数包括第一损失函数和第二损失函数，第一损失函数的值为根据标准图像和样本预测图像所得到的，第二损失函数的值为根据样本预测图像中各图像区域所属的样本类别和标准图像中各图像区域所属的标准类别所得到的。采用本方法能够提高得到的预测图像的准确度。

技术领域

本申请涉及图像预测技术领域，特别是涉及一种图像预测方法、装置、计算机设备和存储介质。

背景技术

随着深度学习技术的发展，深度学习技术逐渐运用在了图像预测或图像生成任务中，例如，通过深度学习技术对降水云图进行预测，从而得到未来一段时间内的降水云图。

传统技术中，图像预测或图像生成通常是把将要生成的图像的每个像素点当成一个回归问题，通过神经网络回归出当前像素点的值，根据回归出的当前像素点的值生成图像。

然而，传统的图像预测方法，存在生成的预测图像准确度较低的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高生成的预测图像准确度的图像预测方法、装置、计算机设备和存储介质。

一种图像预测方法，所述方法包括：

获取待测图像；所述待测图像中包括多个图像区域，且不同的图像区域对应不同的分类类别；

将所述待测图像输入预设的图像预测模型，得到预测图像；其中，所述图像预测模型为根据预设的损失函数训练得到的，所述预设的损失函数包括第一损失函数和第二损失函数，所述第一损失函数的值为根据标准图像和样本预测图像所得到的，所述第二损失函数的值为根据所述样本预测图像中各图像区域所属的样本类别和所述标准图像中各图像区域所属的标准类别所得到的。

在其中一个实施例中，所述图像预测模型的训练过程包括：

将样本图像输入预设的待训练图像预测模型，得到所述样本预测图像；

根据所述标准图像和所述样本预测图像，得到所述第一损失函数的值；

获取所述标准图像中各图像区域所属的标准类别和所述样本预测图像中各图像区域所属的样本类别；

根据所述标准图像中各图像区域所属的标准类别和所述样本预测图像中各图像区域所属的样本类别，得到所述第二损失函数的值；

根据所述第一损失函数的值和所述第二损失函数的值，得到所述待训练图像预测模型的损失函数的值，对所述待训练图像预测模型进行训练，得到所述图像预测模型。

在其中一个实施例中，所述获取所述标准图像中各图像区域所属的标准类别和所述样本预测图像中各图像区域所属的样本类别，包括：

根据预设的转换公式，将所述标准图像中各图像区域的像素值的值域转换为对应的标准概率值，根据所述标准概率值得到所述标准图像中各图像区域所属的标准类别；

根据所述预设的转换公式，将所述样本预测图像中各图像区域的像素值的值域转换为对应的样本概率值，根据所述样本概率值得到所述样本预测图像中各图像区域所属的样本类别。

在其中一个实施例中，所述根据所述第一损失函数的值和所述第二损失函数的值，得到所述待训练图像预测模型的损失函数的值，对所述待训练图像预测模型进行训练，得到所述图像预测模型，包括：

对所述第一损失函数的值和所述第二损失函数的值进行加权求和，得到所述待训练图像预测模型的损失函数的值；