[发明专利]姿态变换数据处理方法、装置、计算机设备和存储介质

说明书

本申请涉及一种姿态变换数据处理方法、装置、计算机设备和存储介质，涉及人工智能的图像处理技术，包括：获取源图像和目标三维姿态，基于语义分割重建得到包括体素的类别信息的三维分割体素，将三维分割体素投影得到对应的目标姿态二维分割图，基于类别信息对目标姿态二维分割图中的对象标注得到部件类别；获取目标三维姿态对应的目标二维姿态，提取源图像、目标姿态二维分割图和目标二维姿态的特征合成中间尺度的变换图像；分别对源图像、三维分割体素、目标二维姿态和变换图像裁切得到各个对象部件的部件层数据，分别对各个对象部件的部件层数据进行部件合成，生成部件图像；将变换图像和部件图像融合得到目标姿态图像，提高姿态变换图像质量。

技术领域

本申请涉及图像处理领域，特别是涉及姿态变换数据处理方法、装置、计算机设备和存储介质。

背景技术

姿态变换是新兴的图像合成研究技术，目的是将给定的对象图像转换到目标姿态，同时维持对象外观的一致性。

相关技术中，可以采用深度学习技术解决姿态变换问题，通过空间变换模块将给定图像特征形变至目标姿态图像，可是当视觉特征因为姿态间隔太大而改变时，将难以得到令人满意的结果，也无法解决遮挡问题，导致姿态变换图像的质量低。

发明内容

基于此，有必要针对上述技术问题，提供一种姿态变换数据处理方法、装置、计算机设备和存储介质，基于语义的对象部件合成高质量的局部部件图像，并且使用带语义分割的三维体素表示来消除二维表示的歧义问题，能够提高目标姿态图像的质量。

一种姿态变换数据处理方法，所述方法包括：获取源图像和目标三维姿态；

结合所述源图像和目标三维姿态，基于语义分割重建得到三维分割体素，所述三维分割体素包括体素的类别信息；将所述三维分割体素投影得到对应的目标姿态二维分割图，基于所述体素的类别信息对所述目标姿态二维分割图中的对象标注得到对应的部件类别；获取所述目标三维姿态对应的目标二维姿态，提取所述源图像、目标姿态二维分割图和目标二维姿态的特征合成中间尺度的变换图像；分别对所述源图像、三维分割体素、目标二维姿态和变换图像裁切得到对应于各个对象部件的部件层数据，所述对象部件是根据所述部件类别确定的；分别对所述各个对象部件的部件层数据进行部件合成，生成各个对象部件对应的部件图像；将所述变换图像和所述部件图像融合得到目标姿态图像。

一种姿态变换数据处理装置，所述装置包括：获取模块，用于获取源图像和目标三维姿态；三维分割体素模块，用于结合所述源图像和目标三维姿态，基于语义分割重建得到三维分割体素，所述三维分割体素包括体素的类别信息；投影模块，用于将所述三维分割体素投影得到对应的目标姿态二维分割图，基于所述体素的类别信息对所述目标姿态二维分割图中的对象标注得到对应的部件类别；中间尺度模块，用于获取所述目标三维姿态对应的目标二维姿态，提取所述源图像、目标姿态二维分割图和目标二维姿态的特征合成中间尺度的变换图像；部件图像生成模块，用于分别对所述源图像、三维分割体素、目标二维姿态和变换图像裁切得到对应于各个对象部件的部件层数据，所述对象部件是根据所述部件类别确定的，分别对所述各个对象部件的部件层数据进行部件合成，生成各个对象部件对应的部件图像；融合模块，用于将所述变换图像和所述部件图像融合得到目标姿态图像。

一种计算机设备，包括存储器和处理器，所述存储器中存储有计算机程序，所述计算机程序被所述处理器执行时，使得所述处理器执行上述姿态变换数据处理方法的步骤。

一种计算机可读存储介质，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时，使得所述处理器执行上述姿态变换数据处理方法的步骤。