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

权利要求书

1.一种图像处理方法，包括：

获取参考图像和目标人员头部图像，其中，所述参考图像包括参考背景和参考人员头部；

利用所述目标人员头部图像替换所述参考图像中参考人员头部，得到待合成图像，其中，所述待合成图像包括部分参考背景、目标人员头部，以及位于所述部分参考背景和所述目标人员头部之间的待填充区域；

对所述参考图像和所述待合成图像进行特征提取，得到肤色样例特征图和填充样例特征图；

基于所述肤色样例特征图、所述填充样例特征图和所述待合成图像生成合成图像。

2.根据权利要求1所述的方法，其中，所述对所述参考图像和所述待合成图像进行特征提取，得到肤色样例特征图和填充样例特征图，包括：

利用特征提取网络提取所述参考图像的特征和所述待合成图像的特征；

采用注意力机制从所述参考图像的特征和所述待合成图像的特征中提取肤色样例特征图和填充样例特征图。

3.根据权利要求2所述的方法，其中，所述采用注意力机制从所提取的所述参考图像和所述待合成图像的特征中提取肤色样例特征图和填充样例特征图，包括：

基于所述待合成图像的特征确定目标人员头部特征和待填充区域特征；

利用所述目标人员头部特征和所述参考图像的特征计算注意力矩阵，得到颜色注意力特征图；

将所述颜色注意力特征图与所述参考图像的特征相乘，得到所述肤色样例特征图；

利用所述待填充区域特征和所述参考图像的特征计算注意力矩阵，得到填充区域注意力特征图；

将所述填充区域注意力特征图与所述参考图像的特征相乘，得到所述填充样例特征图。

4.根据权利要求1-3任一项所述的方法，所述利用所述目标人员头部图像替换所述参考图像中参考人员头部，得到待合成图像，包括：

对所述参考图像中的所述参考人员头部进行五官分割，得到头部掩膜，并将所述参考图像中除所述头部掩膜以外的区域作为所述参考背景；

对所述头部掩膜进行膨胀，得到膨胀区域，其中，所述膨胀区域的面积大于所述目标人员头部图像的面积；

将所述参考背景中与所述膨胀区域不相交的区域确定为所述部分参考背景，将所述目标人员头部图像添加至所述膨胀区域，以及将所述膨胀区域中与所述目标人员头部不相交的区域确定为所述待填充区域；

对所述目标人员头部图像、所述部分参考背景、所述待填充区域进行整合，得到所述待合成图像。

5.根据权利要求4所述的方法，其中，所述基于肤色样例特征图、填充样例特征图和所述待合成图像生成合成图像，包括：

对所述目标人员头部图像进行颜色处理，得到头部灰度图；

对所述肤色样例特征图、所述填充样例特征图、所述头部掩膜、所述头部灰度图以及所述部分参考背景进行合成处理，得到所述合成图像。

6.根据权利要求5所述的方法，其中，所述对所述肤色样例特征图、所述填充样例特征图、所述头部掩膜、所述头部灰度图以及所述部分参考背景进行合成处理，得到所述合成图像，包括：

对所述肤色样例特征图、所述填充样例特征图、所述头部掩膜、所述头部灰度图以及所述部分参考背景进行拼接，得到拼接图；

将所述拼接图输入至预先训练的融合网络中进行融合，得到所述合成图像。

7.根据权利要求1所述的方法，其中，所述参考图像还包括除参考人员头部以外的皮肤裸露区域。

8.一种图像处理装置，包括：

获取模块，被配置为获取参考图像和目标人员头部图像，其中，所述参考图像包括参考背景和参考人员头部；

替换模块，被配置为利用所述目标人员头部图像替换所述参考图像中参考人员头部，得到待合成图像，其中，所述待合成图像包括部分参考背景、目标人员头部，以及位于所述部分参考背景和所述目标人员头部之间的待填充区域；

特征提取模块，被配置为对所述参考图像和所述待合成图像进行特征提取，得到肤色样例特征图和填充样例特征图；

图像生成模块，被配置为基于所述肤色样例特征图、所述填充样例特征图和所述待合成图像生成合成图像。