[发明专利]基于单幅图像的树木建模方法、树木建模装置、设备

权利要求书

1.一种基于单幅图像的树木建模方法，其特征在于，所述树木建模方法包括：

获取待重建单幅树木图像的树木边缘图和可见枝干图；

将所述树木边缘图和所述可见枝干图输入到预先训练好的条件生成对抗网络中，得到树木的轮廓深度图和枝干深度图；

根据所述轮廓深度图得到轮廓点云数据，并根据所述枝干深度图构建得到枝干三维骨架；

将所述轮廓点云数据和所述枝干三维骨架作为约束条件，按照预定的自组织过程式建模方法生成所述待重建单幅树木图像的三维树木模型。

2.根据权利要求1所述的基于单幅图像的树木建模方法，其特征在于，所述待重建单幅树木图像为拍摄得到的真实树木二维图像，所述获取待重建单幅树木图像的树木边缘图的方法包括：

从所述待重建单幅树木图像中分割出树木区域图像；

对所述树木区域图像进行边缘提取，获得所述树木边缘图。

3.根据权利要求2所述的基于单幅图像的树木建模方法，其特征在于，获取待重建单幅树木图像绘制的可见枝干图的方法包括为：

根据用户在显示有所述待重建单幅树木图像的图像编辑器上添加的枝干笔画生成所述可见枝干图。

4.根据权利要求2所述的基于单幅图像的树木建模方法，其特征在于，所述树木建模方法还包括：

生成提醒用户在显示有所述待重建单幅树木图像的图像编辑器上添加的枝干笔画的指令。

5.根据权利要求1所述的基于单幅图像的树木建模方法，其特征在于，所述待重建单幅树木图像为用户绘制的虚拟树木二维图像，获取待重建单幅树木图像的树木边缘图和可见枝干图的方法包括：

根据用户在图像编辑器上绘制待重建单幅树木图像时的绘制笔画生成树木边缘图和可见枝干图。

6.根据权利要求1所述的基于单幅图像的树木建模方法，其特征在于，预训练的所述条件生成对抗网络包括预训练的第一条件生成对抗网络和预训练的第二条件生成对抗网络，将所述树木边缘图和所述可见枝干图输入到预先训练好的条件生成对抗网络中，得到轮廓深度图和枝干深度图的方法为：将所述树木边缘图输入到所述第一条件生成对抗网络，得到轮廓深度图，将所述可见枝干图输入到所述第二条件生成对抗网络，得到枝干深度图。

7.一种基于单幅图像的树木建模装置，其特征在于，所述树木建模装置包括：

获取单元，用于获取待重建单幅树木图像的树木边缘图和可见枝干图；

预训练的条件生成对抗网络，用于根据所述树木边缘图和所述可见枝干图得到轮廓深度图和枝干深度图；

恢复单元，用于根据所述轮廓深度图得到轮廓点云数据并根据所述枝干深度图构建得到枝干三维骨架；

重建单元，用于在所述轮廓点云数据和所述枝干三维骨架的约束条件下，按照预定的自组织过程式建模方法生成所述待重建单幅树木图像的三维树木模型。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有基于单幅图像的树木建模程序，所述基于单幅图像的树木建模程序被处理器执行时实现权利要求1至6任一项所述的基于单幅图像的树木建模方法。

9.一种计算机设备，其特征在于，所述计算机设备包括计算机可读存储介质、处理器和存储在所述计算机可读存储介质中的基于单幅图像的树木建模程序，所述基于单幅图像的树木建模程序被处理器执行时实现权利要求1至6任一项所述的基于单幅图像的树木建模方法。