[发明专利]一种色斑预测方法、装置、设备及存储介质

权利要求书

1.一种色斑预测方法，其特征在于，所述方法包括：

获取待预测图像；

将所述待预测图像输入至预先训练得到的色斑预测模型中进行预测处理，所述色斑预测模型为全卷积生成对抗网络模型；

通过所述色斑预测模型得到色斑预测结果图。

2.如权利要求1所述的方法，其特征在于，所述将所述待预测图像输入至预先训练得到的色斑预测模型中进行预测处理之前，还包括：

确定所述待预测图像中的色斑信息，所述色斑信息包括：色斑的位置以及类别；

根据所述待预测图像中的色斑信息，对所述待预测图像进行预处理，得到预处理后的多帧图像，所述多帧图像分别包括：无色斑的图像以及标识有色斑类别的图像；

所述将所述待预测图像输入至预先训练得到的色斑预测模型中进行预测处理，包括：

将所述预处理后的多帧图像输入至预先训练得到的色斑预测模型中进行预测处理。

3.如权利要求1或2所述的方法，其特征在于，所述将所述待预测图像输入至预先训练得到的色斑预测模型中进行预测处理之前，还包括：

获取目标待处理图像，所述目标待处理图像中包括色斑信息；

根据所述目标待处理图像分别确定多个目标通道图像，所述目标通道图像包括：去除色斑信息的多通道图像以及色斑类别通道图像；

将多个所述目标通道图像以及目标噪声图像合并，一起输入待训练神经网络结构中得到训练后的色斑预测模型；所述目标噪声图像为随机生成的噪声图像。

4.如权利要求3所述的方法，其特征在于，所述根据所述目标待处理图像分别确定多个目标通道图像，包括：

确定所述目标待处理图像中的色斑信息；

根据所述目标待处理图像中的色斑信息，对所述目标待处理图像进行色斑去除处理，得到去除色斑信息的目标待处理图像。

5.如权利要求3所述的方法，其特征在于，所述根据所述目标待处理图像分别确定多个目标通道图像，包括：

对所述目标待处理图像进行色斑检测处理，得到所述色斑类别通道图像。

6.如权利要求5所述的方法，其特征在于，所述对所述目标待处理图像进行色斑检测处理，得到所述色斑类别通道图像，包括：

分别确定所述目标待处理图像中每种类型的色斑信息的位置；

在所述色斑信息的位置处设置所述色斑信息对应类型的灰度信息，得到所述色斑类别通道图像。

7.如权利要求3所述的方法，其特征在于，所述将多个所述目标通道图像以及目标噪声图像合并，一起输入待训练神经网络结构中得到训练后的色斑预测模型之前，所述方法包括：

分别将所述目标通道图像以及所述目标噪声图像进行归一化处理，得到目标输入图像；

所述将多个所述目标通道图像以及目标噪声图像分别输入待训练神经网络结构中得到训练后的色斑预测模型，包括：

将所述目标输入图像输入待训练神经网络结构中得到训练后的色斑预测模型。

8.一种色斑预测装置，其特征在于，所述装置包括：获取模块、预测模块、输出模块；

所述获取模块，用于获取待预测图像；

所述预测模块，用于将所述待预测图像输入至预先训练得到的色斑预测模型中进行预测处理，所述色斑预测模型为全卷积生成对抗网络模型；

所述输出模块，用于通过所述色斑预测模型得到色斑预测结果图。

9.一种计算机设备，其特征在于，包括：存储器、处理器，所述存储器中存储有可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时，实现上述权利要求1至7任一项所述的方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述存储介质上存储有计算机程序，该计算机程序被处理器执行时，实现权利要求1至7中任一项所述方法的步骤。