[发明专利]发丝线段处理模型的训练方法、装置和发丝线段处理方法

说明书

本公开提供了一种发丝线段处理模型的训练方法、装置和发丝线段处理方法，涉及人工智能技术领域，尤其涉及计算机视觉、增强现实、虚拟现实、深度学习等技术领域。该方法包括：获取样本发丝线段在三维空间的标准方向和位置坐标；基于标准方向获取样本发丝线段的初始方向，并将初始方向和位置坐标输入发丝线段处理模型中；对初始方向和位置坐标进行编码，生成样本发丝线段的隐向量空间，并对隐向量空间进行解码，获取样本发丝线段在三维空间的预测方向；基于标准方向和预测方向对发丝线段处理模型进行调整，得到目标发丝线段处理模型。本公开大幅增强了发丝线段处理模型的性能，可以提升发丝模型生产效率和发丝生成效果。

技术领域

本公开涉及人工智能技术领域，尤其涉及计算机视觉、增强现实、虚拟现实、深度学习等技术领域，可应用于元宇宙、数字人等场景。

背景技术

相关技术中，三维虚拟形象在社交、直播、游戏等元宇宙、数字人场景具有广泛应用价值。通过多视角发型图像生成虚拟形象可以有效满足用户的个性化需求。虚拟形象中发丝建模复杂，发丝数量大，由于发丝线段缺乏走向，发丝预测经常出现杂乱的情况。

因此，如何提升发丝模型生产效率，提升发丝生成效果，降低三维形象构建成本，已经成为重要的研究方向之一。

发明内容

本公开提供了一种发丝线段处理模型的训练方法、装置和发丝线段处理方法。

根据本公开的一方面，提供了一种发丝线段处理模型的训练方法，该方法包括：

获取样本发丝线段的训练数据，训练数据包括样本发丝线段在三维空间的标准方向和位置坐标；

基于标准方向获取样本发丝线段的初始方向，并将初始方向和位置坐标输入发丝线段处理模型中；

对初始方向和位置坐标进行编码，生成样本发丝线段的隐向量空间，并对隐向量空间进行解码，获取样本发丝线段在三维空间的预测方向；

基于标准方向和预测方向对发丝线段处理模型进行调整，得到目标发丝线段处理模型。

本公开大幅增强了发丝线段处理模型的性能，可以基于数据性质实现发丝线段处理模型的强化，可以提升发丝模型生产效率，提升发丝生成效果，降低三维形象构建成本。

根据本公开的一方面，提供了一种发丝线段处理方法，该方法包括：

获取多视角发型图像，并基于多视角发型图像获取三维空间下的多个发丝线段的位置坐标和初始方向；

针对多个发丝线段中的任一发丝线段，将发丝线段的位置坐标和初始方向输入目标发丝线段处理模型中，由目标发丝线段处理模型输出发丝线段在三维空间下的预测方向；

其中，目标发丝线段处理模型为采用如上述第一方面实施例的训练方法训练得到的模型。

本公开实施例中，可以获取发型线段的有向方向，减少了发丝预测存在的歧义性，令发型生成的发丝更加均匀和整齐，提升了发丝生成的效果，还可以提升发丝模型的生产效率。

根据本公开的另一方面，提供了一种发丝线段处理模型的训练装置，包括：

获取模块，用于获取样本发丝线段的训练数据，训练数据包括样本发丝线段在三维空间的标准方向和位置坐标；

输入模块，用于基于标准方向获取样本发丝线段的初始方向，并将初始方向和位置坐标输入发丝线段处理模型中；

处理模块，用于对初始方向和位置坐标进行编码，生成样本发丝线段的隐向量空间，并对隐向量空间进行解码，获取样本发丝线段在三维空间的预测方向；

调整模块，用于基于标准方向和预测方向对发丝线段处理模型进行调整，得到目标发丝线段处理模型。

根据本公开的另一方面，提供了一种发丝线段处理装置，包括：