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

权利要求书

1.一种发丝线段处理模型的训练方法，其中，包括：

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

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

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

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

2.根据权利要求1所述的方法，其中，所述发丝线段处理模型包括无向线段编码器，所述对所述初始方向和所述位置坐标进行编码，生成所述样本发丝线段的隐向量空间，包括：

由所述无向线段编码器中的第一编码层，对所述位置坐标进行位置编码，获取位置特征表示；

由所述无向线段编码器中的多层感知机，对所述初始方向进行解析；

由所述无向线段编码器中的第二编码层，对解析后的所述初始方向和所述位置特征表示进行共同编码，以生成所述隐向量空间。

3.根据权利要求2所述的方法，其中，所述发丝线段处理模型包括有向线段解码器，所述对所述隐向量空间进行解码，获取所述样本发丝线段在三维空间的预测方向，包括：

由所述有向线段解码器对所述隐向量空间进行多层感知，获取所述发丝线段在三维空间的所述预测方向。

4.根据权利要求1-3任一项所述的方法，其中，所述基于所述标准方向和所述预测方向对所述发丝线段处理模型进行调整，包括：

根据所述标准方向和所述预测方向，确定所述发丝线段处理模型的损失函数；

基于所述损失函数对所述发丝线段处理模型进行调整，获取所述调整后的发丝线段处理模型。

5.根据权利要求4所述的方法，其中，所述基于所述标准方向获取所述样本发丝线段的初始方向，包括：

对所述样本发丝线段的标准方向进行随机反转，并将随机反转后的方向作为所述样本发丝线段对应的所述初始方向。

6.一种发丝线段处理方法，其中，包括：

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

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

其中，所述目标发丝线段处理模型为采用如权利要求1-5中任一项所述的训练方法训练得到的模型。

7.根据权利要求6所述的方法，其中，所述基于所述多视角发型图像获取三维空间下的多个发丝线段的位置坐标和初始方向，包括：

调用三维重建工具对所述多视角发型图像进行帧间位姿估计，获取全局位姿三维数据；

调用动作捕捉系统对所述全局位姿三维数据进行头部位姿估计，获取头部位姿三维数据；

对所述头部位姿三维数据进行神经渲染重建，并提取三维空间下的所述多个发丝线段的位置坐标和初始方向。

8.根据权利要求6所述的方法，其中，还包括：

基于所述多个发丝线段的位置坐标和预测方向进行可微发丝拟合，以获取所述多视角发型图像对应的发型模型。

9.一种发丝线段处理模型的训练装置，其中，包括：

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

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

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

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