图像处理方法及装置、图像生成模型的训练方法及装置.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN114187633A(43)申请公布日2022.03.15(21)申请号202111487641.0G06N3/08(2006.01)(22)申请日2021.12.07G06T11/00(2006.01)(71)申请人北京百度网讯科技有限公司地址100085北京市海淀区上地十街10号百度大厦2层(72)发明人彭昊天(74)专利代理机构中科专利商标代理有限责任公司11021代理人张霞(51)Int.Cl.G06V40/16(2022.01)G06V10/774(2022.01)G06V10/82(2022.01)G06K9/62(2022.01)G06N3/04(2006.01)权利要求书6页说明书12页附图6页(54)发明名称图像处理方法及装置、图像生成模型的训练方法及装置(57)摘要本公开提供了一种图像处理方法及装置、图像生成模型的训练方法及装置、设备、介质和产品，涉及人工智能领域，尤其涉及计算机视觉、虚拟/增强现实和图像处理技术领域。具体实现方案包括：对目标图像进行识别，以提取目标图像中对象的面部检测区域，面部检测区域包括对象的面部区域和头发区域；确定与头发区域关联的发丝布局特征，发丝布局特征包括平面布局特征和/或空间布局特征；以及根据发丝布局特征，确定与头发区域匹配的虚拟发型图像，以实现利用虚拟发型图像替换目标图像中的头发区域。CN114187633ACN114187633A权利要求书1/6页1.一种图像处理方法，包括：对目标图像进行识别，以提取所述目标图像中对象的面部检测区域，所述面部检测区域包括对象的面部区域和头发区域；确定与所述头发区域关联的发丝布局特征，所述发丝布局特征包括平面布局特征和/或空间布局特征；以及根据所述发丝布局特征，确定与所述头发区域匹配的虚拟发型图像，以利用所述虚拟发型图像替换所述目标图像中的所述头发区域。2.根据权利要求1所述的方法，还包括，在对目标图像进行识别之前：提取所述目标图像中的面部特征点；根据所述目标图像中的面部特征点和预设基准占位特征，对所述目标图像进行图像配准，得到配准后的目标图像，其中，所述对目标图像进行识别，以提取所述目标图像中对象的面部检测区域，包括：对所述配准后的目标图像进行识别，以提取所述配准后的目标图像中对象的面部检测区域，其中，所述基准占位特征指示了对象的面部区域在所在图像中的参考占位。3.根据权利要求1所述的方法，其中，所述发丝布局特征包括平面布局特征，所述确定与所述头发区域关联的发丝布局特征，包括：确定所述面部检测区域中的面部区域的轮廓信息，得到与所述头发区域关联的内轮廓特征；确定所述面部检测区域和除所述面部检测区域以外的背景区域的交界信息，得到与所述头发区域关联的外轮廓特征，以及所述内轮廓特征和所述外轮廓特征构成所述平面布局特征。4.根据权利要求3所述的方法，其中，所述根据所述发丝布局特征，确定与所述头发区域匹配的虚拟发型图像，包括：确定与所述头发区域的内轮廓特征和外轮廓特征的对应相似度均高于预设阈值的第一虚拟发型图像；以及基于所述第一虚拟发型图像，确定与所述头发区域匹配的虚拟发型图像。5.根据权利要求4所述的方法，其中，所述确定与所述头发区域关联的发丝布局特征，还包括：确定所述头发区域中的像素的梯度信息；根据所述头发区域中的像素的梯度信息，确定与所述头发区域关联的发丝走向特征；以及与所述头发区域关联的发丝走向特征构成所述平面布局特征，其中，所述基于所述第一虚拟发型图像，确定与所述头发区域匹配的虚拟发型图像，包括：在所述第一虚拟发型图像中，确定与所述发丝走向特征的相似度高于预设阈值的第二虚拟发型图像，作为与所述头发区域匹配的虚拟发型图像。6.根据权利要求4所述的方法，其中，所述发丝布局特征还包括空间布局特征，所述确定与所述头发区域关联的发丝布局特征，还包括：2CN114187633A权利要求书2/6页筛选所述头发区域中符合预设条件的有效角点；计算与所述有效角点关联的特征描述符，所述特征描述符指示了所述有效角点与所述目标图像中的人像中心点的距离和夹角信息；根据与所述有效角点关联的特征描述符，确定与所述头发区域关联的发丝分布的空间布局特征，其中，所述基于所述第一虚拟发型图像，确定与所述头发区域匹配的虚拟发型图像，还包括：在所述第一虚拟发型图像中，确定与所述发丝分布的空间布局特征的相似度高于预设阈值的第三虚拟发型图像，作为与所述头发区域匹配的虚拟发型图像。7.根据权利要求1至6中任一项所述的方法，其中，利用生成式对抗网络，将与所述目标图像中的头发区域关联的发丝布局特征作为约束，输出与所述头发区域匹配的虚拟发型图像。8.一种图像生成模型的训练方法，包括：对样本图像进行识别，以提取所述样本图像中