[发明专利]一种面向家装设计的基于生成对抗网络技术的增强现实图像渲染上色方法

权利要求书

1.一种面向家装设计的基于生成对抗网络技术的增强现实图像上色方法，其特征在于，包括以下步骤：

S101：采集实时视频；

S102：扫描数字化标识物；

S103：通过增强现实程序识别标识物；

S104：标识物与三维虚拟物体进行匹配；

S105：根据标识物位置调整三维虚拟模型的位置；

S106：确定风格要求；

S107：匹配预训练上色模型库；

S108：虚拟物体与视频流背景融合；

S109：虚拟物体上色入视频流。

2.根据权利要求1所述的一种面向家装设计的基于生成对抗网络技术的增强现实图像上色方法，其特征在于，所述的匹配预训练上色模型库的方法一包括以下步骤：

S201：输入需要上色的三维模型顶点坐标；

S202：将三维模型放到三维场景核实的位置；

S203：设置摄像头的角度和视角；

S204：设置光照的位置、颜色、方向参数；

S205：设置三维模型的颜色参数；

S206：将上色后的模型输入生成对抗网络模型；

S207：将通过判别网络的模型存储至预训练模型库。

3.根据权利要求2所述的一种面向家装设计的基于生成对抗网络技术的增强现实图像上色方法，其特征在于，所述的将上色后的模型输入生成对抗网络模型，包括以下步骤：

S301：输入已经上色后的三维模型；

S302：存储至判别网络模型库，判别网络存储的是设置好参数的上色模型；

S303：生成网络输出生成后的三维模型；

S304：判别网络计算出生成模型和模型库的相似值；

S305：生成模型和模型库的相似值如果大于等于预设的阈值，则判定生成上色的三维模型接近真实的模型；相似值如果小于预设的阈值，则判定生成网络上色的三维模型非真实的模型；重复S303，S304步骤直到判别网络给出生成后的上色模型为真实；

S306：输出通过判别网络的三维上色模型。

4.根据权利要求1所述的一种面向家装设计的基于生成对抗网络技术的增强现实图像上色方法，其特征在于，所述的匹配预训练上色模型库的方法二包括以下步骤：

S401：输入已知不同作者不同类型图像；

S402：将输入的图像生成对抗网络模型；

S403：将通过判别网络的模型存储至预训练模型库。

5.根据权利要求1所述的一种面向家装设计的基于生成对抗网络技术的增强现实图像上色方法，其特征在于，所述的将输入的图像生成对抗网络模型，包括以下步骤：

S501：输入成品图像模型；

S502：存储至判别网络模型库，判别网络存储的是设置好参数的上色模型；

S503：生成网络输出生成后的图像模型；

S504：判别网络计算出生成模型和模型库的相似值；

S505：生成模型和模型库的相似值如果大于等于预设的阈值，则判定生成上色的图像模型接近真实的模型；相似值如果小于预设的阈值，则判定生成网络上色的图像模型非真实的模型，重复S503，S504步骤直到判别网络给出生成后的上色模型为真实。

6.根据权利要求1所述的一种面向家装设计的基于生成对抗网络技术的增强现实图像上色方法，其特征在于，所述的虚拟物体与视频流背景融合，包括以下步骤：

S601：使用识别程序识别出视频流背景物体轮廓；

S602：提取视频流背景物体位置坐标坐标；

S603：以位置坐标为参考点将虚拟物体叠加显示在视频流背景物体上。