[发明专利]图像处理方法、装置、设备及介质

权利要求书

1.一种图像处理方法，其特征在于，包括：

获取线稿图；

获取对所述线稿图中的目标部位进行上色的目标颜色；

提取所述线稿图的线稿图特征；

提取所述目标颜色的目标颜色特征；

将所述目标颜色特征与所述线稿图特征进行特征融合，以生成融合后的线稿图特征；

基于所述融合后的线稿图特征，生成所述线稿图的上色图像，其中，所述上色图像中的所述目标部位的颜色与所述目标颜色一致。

2.根据权利要求1所述的方法，其特征在于，所述获取对所述线稿图中的目标部位进行上色的目标颜色，包括：

获取用户确定的所述目标部位；

获取所述用户确定的用于对所述目标部位进行上色的目标颜色。

3.根据权利要求1所述的方法，其特征在于，所述提取所述目标颜色的目标颜色特征，包括：

根据所述线稿图特征的特征维度对所述目标颜色进行编码，以获取与所述线稿图特征的特征维度一致的目标颜色特征。

4.根据权利要求1-3中任一项所述的方法，其特征在于，所述特征融合由用于基于参考颜色对线稿图进行上色的模型执行。

5.根据权利要求4所述的方法，其特征在于，还包括：

训练所述用于基于参考颜色对线稿图进行上色的模型。

6.根据权利要求5所述的方法，其特征在于，所述训练用于基于参考颜色对线稿图进行上色的模型，包括：

获取与样本图像对应的样本线稿图；

提取所述样本线稿图的样本线稿图特征；

对所述样本图像进行语义分割处理，以获取样本部位；

对所述样本部位的所有像素进行平均，以获取样本颜色；

提取所述样本颜色的样本颜色特征；

将所述样本颜色特征与所述线稿图特征进行特征混合，以生成融合后的样本线稿图特征；

对所述融合后的样本线稿图特征进行解码，以生成所述样本线稿图的样本上色图像；

根据所述样本线稿图、所述样本图像和所述样本上色图像来训练所述模型的参数。

7.根据权利要求6所述的方法，其特征在于，所述根据所述样本线稿图、所述样本图像和所述样本上色图像来训练所述模型的参数，包括：

根据所述样本上色图像和所述样本图像生成目标损失函数；以及

基于所述样本线稿图、所述样本图像和所述样本上色图像，根据所述目标损失函数的反向传播，训练所述模型的参数。

8.根据权利要求7所述的方法，其特征在于，所述根据所述样本上色图像和所述样本图像生成目标损失函数，包括：

计算所述样本上色图像中每个像素值和所述样本图像中每个像素值之间的平均绝对误差，获取重建损失函数；和/或，

计算所述样本上色图像中每个像素值和所述样本图像中每个像素值之间的均方误差，获取风格损失函数；和/或，

根据预设的判别器模型对所述样本上色图像和所述样本图像进行分类，获取对抗损失函数。

9.根据权利要求1-8中任一项所述的方法，其特征在于，所述目标部位包括：头发、眼睛、嘴巴、面部、脖子和/或衣服。

10.一种图像处理装置，其特征在于，包括：

第一获取模块，用于获取线稿图；

第二获取模块，用于获取对所述线稿图中的目标部位进行上色的目标颜色；

第一提取模块，用于提取所述线稿图的线稿图特征；

第二提取模块，用于提取所述目标颜色的目标颜色特征；

融合模块，用于将所述目标颜色特征与所述线稿图特征进行特征融合，以生成融合后的线稿图特征；

生成模块，用于基于所述融合后的线稿图特征，生成所述线稿图的上色图像，其中，所述上色图像中的所述目标部位的颜色与所述目标颜色一致。