工业零件的CAD模型生成方法及装置

本发明涉及计算机，尤其涉及一种工业零件的cad模型生成方法及装置。

背景技术：

1、在计算机辅助设计（computer-aided design，cad）和产品设计领域，从二维图像或草图快速生成三维cad模型是提升设计效率的关键技术之一。三维cad模型广泛应用于汽车、航空航天、消费电子、模具设计等多个行业，其生成效率直接影响到产品研发周期。在众多三维表示方法中，边界表示法（boundary representation，b-rep）是现代cad系统的核心数据格式，它通过精确的几何面、边和顶点来定义物体的边界，能够准确描述具有复杂形状的实体模型。

2、现有的图像到cad生成方法通常需要分别训练条件模型和无条件模型，或者只能执行单一的重建任务，导致同一个模型无法同时满足“精确重建”和“创意生成”两种需求，用户在不同设计目标之间切换时不得不更换模型或训练策略，使得控制能力不足，导致极大降低了工作效率。

技术实现思路

1、本发明提供一种工业零件的cad模型生成方法及装置，用以解决现有技术中控制能力不足，导致极大降低了工作效率的问题。

2、本发明提供一种工业零件的cad模型生成方法，包括：

3、将工业零件图像输入至视觉特征提取模型，得到所述视觉特征提取模型输出的图像条件特征信息；

4、基于所述图像条件特征信息、无条件特征信息和引导尺度参数，确定组合特征信息；

5、基于所述组合特征信息，生成所述工业零件图像对应的cad模型。

6、根据本发明提供的一种工业零件的cad模型生成方法，在所述组合特征信息包括第一组合特征信息的情况下，所述基于所述图像条件特征信息、无条件特征信息和引导尺度参数，确定组合特征信息，包括：

7、在拓扑生成阶段，基于所述图像条件特征信息、所述无条件特征信息对应的偏移特征信息和所述引导尺度参数，确定所述第一组合特征信息。

8、根据本发明提供的一种工业零件的cad模型生成方法，所述基于所述图像条件特征信息、所述无条件特征信息对应的偏移特征信息和所述引导尺度参数，确定所述第一组合特征信息，包括：

9、基于所述引导尺度参数，对所述图像条件特征信息和所述无条件特征信息对应的偏移特征信息进行加权组合，得到所述第一组合特征信息。

10、根据本发明提供的一种工业零件的cad模型生成方法，在所述组合特征信息包括第二组合特征信息的情况下，所述基于所述图像条件特征信息、无条件特征信息和引导尺度参数，确定组合特征信息，包括：

11、在几何生成阶段，基于所述图像条件特征信息、所述无条件特征信息和所述引导尺度参数，确定所述第二组合特征信息。

12、根据本发明提供的一种工业零件的cad模型生成方法，所述基于所述图像条件特征信息、所述无条件特征信息和所述引导尺度参数，确定所述第二组合特征信息，包括：

13、基于所述引导尺度参数，对所述图像条件特征信息和所述无条件特征信息进行加权组合，得到所述第二组合特征信息。

14、根据本发明提供的一种工业零件的cad模型生成方法，所述基于所述组合特征信息，生成所述工业零件图像对应的cad模型，包括：

15、基于所述组合特征信息中的第一组合特征信息和第二组合特征信息，生成所述工业零件图像对应的cad模型。

16、根据本发明提供的一种工业零件的cad模型生成方法，所述基于所述组合特征信息中的第一组合特征信息和第二组合特征信息，生成所述工业零件图像对应的cad模型，包括：

17、将所述第一组合特征信息输入至第一模型，得到所述第一模型输出的离散拓扑结构属性信息；

18、基于所述离散拓扑结构属性信息和所述第二组合特征信息，生成所述工业零件图像对应的cad模型。

19、根据本发明提供的一种工业零件的cad模型生成方法，所述基于所述离散拓扑结构属性信息和所述第二组合特征信息，生成所述工业零件图像对应的cad模型，包括：

20、将所述离散拓扑结构属性信息和所述第二组合特征信息输入至第二模型，得到所述第二模型输出的连续几何属性信息；

21、基于所述离散拓扑结构属性信息和所述连续几何属性信息，生成所述工业零件图像对应的cad模型。

22、根据本发明提供的一种工业零件的cad模型生成方法，所述偏移特征信息是基于所述无条件特征信息和偏移量确定的。

23、根据本发明提供的一种工业零件的cad模型生成方法，所述偏移量由随机潜在变量生成。

24、本发明还提供一种工业零件的cad模型生成装置，包括：

25、特征提取模块，用于将工业零件图像输入至视觉特征提取模型，得到所述视觉特征提取模型输出的图像条件特征信息；

26、确定模块，用于基于所述图像条件特征信息和无条件特征信息，确定组合特征信息；

27、生成模块，用于基于所述组合特征信息，生成所述工业零件图像对应的cad模型。

28、本发明还提供一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如上述任一种所述工业零件的cad模型生成装置方法。

29、本发明还提供一种非暂态计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现如上述任一种所述工业零件的cad模型生成装置方法。

30、本发明还提供一种计算机程序产品，包括计算机程序，所述计算机程序被处理器执行时实现如上述任一种所述工业零件的cad模型生成装置方法。

31、本发明提供的工业零件的cad模型生成方法及装置，通过将工业零件图像输入至视觉特征提取模型，得到所述视觉特征提取模型输出的图像条件特征信息；基于所述图像条件特征信息、无条件特征信息和引导尺度参数，确定组合特征信息；基于所述组合特征信息，生成所述工业零件图像对应的cad模型。通过图像条件特征信息、无条件特征信息和引导尺度参数确定的组合特征信息，能够自动控制实现从无条件随机生成到高保真条件重建的平滑过程，进而实现工业零件图像对应的cad模型的准确生成，提升工作效率。

1.一种工业零件的计算机辅助设计cad模型生成方法，其特征在于，包括：

2.根据权利要求1所述的工业零件的cad模型生成方法，其特征在于，在所述组合特征信息包括第一组合特征信息的情况下，所述基于所述图像条件特征信息、无条件特征信息和引导尺度参数，确定组合特征信息，包括：

3.根据权利要求2所述的工业零件的cad模型生成方法，其特征在于，所述基于所述图像条件特征信息、所述无条件特征信息对应的偏移特征信息和所述引导尺度参数，确定所述第一组合特征信息，包括：

4.根据权利要求1所述的工业零件的cad模型生成方法，其特征在于，在所述组合特征信息包括第二组合特征信息的情况下，所述基于所述图像条件特征信息、无条件特征信息和引导尺度参数，确定组合特征信息，包括：

5.根据权利要求4所述的工业零件的cad模型生成方法，其特征在于，所述基于所述图像条件特征信息、所述无条件特征信息和所述引导尺度参数，确定所述第二组合特征信息，包括：

6.根据权利要求1所述的工业零件的cad模型生成方法，其特征在于，所述基于所述组合特征信息，生成所述工业零件图像对应的cad模型，包括：

7.根据权利要求6所述的工业零件的cad模型生成方法，其特征在于，所述基于所述组合特征信息中的第一组合特征信息和第二组合特征信息，生成所述工业零件图像对应的cad模型，包括：

8.根据权利要求7所述的工业零件的cad模型生成方法，其特征在于，所述基于所述离散拓扑结构属性信息和所述第二组合特征信息，生成所述工业零件图像对应的cad模型，包括：

9.根据权利要求2所述的工业零件的cad模型生成方法，其特征在于，所述偏移特征信息是基于所述无条件特征信息和偏移量确定的。

10.根据权利要求9所述的工业零件的cad模型生成方法，其特征在于，所述偏移量由随机潜在变量生成。

11.一种工业零件的cad模型生成装置，其特征在于，包括：

12.根据权利要求11所述的工业零件的cad模型生成装置，其特征在于，在所述组合特征信息包括第一组合特征信息的情况下，所述确定模块，具体用于：