一种三维模型检索方法、装置及计算机可读存储介质与流程

本发明涉及智能检索，尤其涉及一种三维模型检索方法、装置、电子设备及计算机可读存储介质。

背景技术：

1、三维模型数据库是存储有制造业相关的工业三维模型的数据库。智能检索是三维模型数据库的必备技能，用户可以通过上传3d(三维)模型文件从三维模型数据库中的海量三维模型中检索到需要的模型，从而大幅减少用户开发或选型时间，提高工作效率。

2、然而，随着海量三维模型出现，三维模型数据库中的三维模型数量越来越多，传统基于拓扑结构的三维检索方法在检索性能和精度上无法满足工程师的需求，存在检索速度慢以及检索结果不准确的问题。

技术实现思路

1、为了解决上述提出的至少一个技术问题，本发明提供一种三维模型检索方法、装置、电子设备及计算机可读存储介质。

2、第一方面，提供了一种三维模型检索方法，所述方法：获取待进行模型检索的三维模型文件；基于预训练好的人工智能模型对所述三维模型文件进行特征提取，得到所述三维模型文件的特征向量；根据所述特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型，所述三维模型数据库为存储有三维模型的数据库。

3、在该方面中，通过预训练好的人工智能模型提取待进行模型检索的三维模型文件的特征向量，并根据特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型，实现了三维模型的检索，且提高了检索精度和检索效率。

4、在一种可能实现的方式中，所述人工智能模型的预训练包括：

5、根据所述三维模型数据库中存储的三维模型，采用目标训练方法训练所述人工智能模型，所述目标训练方法包括分层分类和度量学习。

6、在该种可能实现的方式中，采用的分层分类和度量学习这两种目标训练方法在机器学习中具有降低所训练的模型复杂性和提高所训练的模型运算准确性的作用。

7、在一种可能实现的方式中，所述根据所述三维模型数据库中存储的三维模型，采用目标训练方法训练所述人工智能模型，包括：

8、对所述三维模型数据库中存储的三维模型进行预处理操作，所述预处理操作包括多视图提取、数据格式转换和三维模型位置矫正；

9、将预处理后的所述三维模型作为输入，采用所述目标训练方法训练所述人工智能模型。

10、在该种可能实现的方式中，对三维模型数据库中存储的三维模型进行预处理操作后，提高三维模型作为人工智能模型的输入的适配性，同时也能提高模型训练的效率和后续模型运算的准确性。

11、在一种可能实现的方式中，在所述基于预训练好的人工智能模型对所述三维模型文件进行特征提取，得到所述三维模型文件的特征向量之前，所述三维模型检索方法还包括；

12、对所述三维模型文件进行所述预处理操作。

13、在该种可能实现的方式中，对待进行检索的三维模型文件进行上述的预处理操作，提高了对三维模型文件的特征提取效率和准确性，从而提高了后续基于所提取的特征向量的模型检索结果更加准确。

14、在一种可能实现的方式中，所述三维模型数据库包括第一数据库和第二数据库，所述第一数据库存储有各所述三维模型的缩略图，所述第二数据库存储有各个所述三维模型对应的个体信息；

15、所述根据所述特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型，包括：

16、根据所述特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到并显示相似性最高的前预设数量个所述目标三维模型的缩略图及对应的个体信息。

17、在该种可能实现的方式中，通过模型检索得到并显示相似性最高的前预设数量个所述目标三维模型的缩略图及对应的个体信息，提高了检索结果的直观性和内容的丰富性，提升用户的检索体验。

18、在一种可能实现的方式中，所述三维模型检索方法还包括：

19、基于预训练好的所述人工智能模型识别所述三维模型文件中的三维模型的类别；

20、在所述根据所述特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型之后，所述三维模型检索方法还包括：

21、将预设数量个所述目标三维模型中模型类别与所述三维模型文件中的三维模型的类别不同的三维模型剔除。

22、在该种可能实现的方式中，通过识别三维模型文件中的三维模型的类别，并将检索结果中类别与所述三维模型文件中的三维模型的类别不同的三维模型剔除，提高了三维模型检索结果的准确性。

23、在一种可能实现的方式中，在所述基于预训练好的人工智能模型对所述三维模型文件进行特征提取，得到所述三维模型文件的特征向量之前，所述三维模型检索方法还包括：

24、对所述三维模型文件进行数据有效性校验。

25、在该种可能实现的方式中，通过对三维模型文件进行数据有效性校验，能够拦截非法的、无效的输入，提高了安全性并避免了不必要的模型检索工作。

26、第二方面，提供了一种三维模型检索装置，所述装置包括：

27、获取模块，用于获取待进行模型检索的三维模型文件；

28、提取模块，用于基于预训练好的人工智能模型对所述三维模型文件进行特征提取，得到所述三维模型文件的特征向量；

29、检索模块，用于根据所述特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型，所述三维模型数据库为存储有三维模型的数据库。

30、第三方面，提供了一种电子设备，包括：处理器、发送装置、输入装置、输出装置和存储器，所述存储器用于存储计算机程序代码，所述计算机程序代码包括计算机指令，当所述处理器执行所述计算机指令时，所述电子设备执行如上述的三维模型检索方法。

31、第四方面，提供了一种计算机可读存储介质，所述计算机可读存储介质中存储有计算机程序，所述计算机程序包括程序指令，所述程序指令当被电子设备的处理器执行时，使所述处理器执行如上述的三维模型检索方法。

32、应当理解的是，以上的一般描述和后文的细节描述仅是示例性和解释性的，而非限制本公开。

技术特征：

1.一种三维模型检索方法，其特征在于，包括如下步骤：

2.根据权利要求1所述的三维模型检索方法，其特征在于，所述人工智能模型的预训练包括：

3.根据权利要求2所述的三维模型检索方法，其特征在于，所述根据所述三维模型数据库中存储的三维模型，采用目标训练方法训练所述人工智能模型，包括：

4.根据权利要求3所述的三维模型检索方法，其特征在于，在所述基于预训练好的人工智能模型对所述三维模型文件进行特征提取，得到所述三维模型文件的特征向量之前，所述三维模型检索方法还包括；

5.根据权利要求1所述的三维模型检索方法，其特征在于，所述三维模型数据库包括第一数据库和第二数据库，所述第一数据库存储有各所述三维模型的缩略图，所述第二数据库存储有各个所述三维模型对应的个体信息；

6.根据权利要求1所述的三维模型检索方法，其特征在于，还包括：

7.根据权利要求1所述的三维模型检索方法，其特征在于，在所述基于预训练好的人工智能模型对所述三维模型文件进行特征提取，得到所述三维模型文件的特征向量之前，所述三维模型检索方法还包括：

8.一种三维模型检索装置，其特征在于，包括：

9.一种电子设备，其特征在于，包括：处理器、发送装置、输入装置、输出装置和存储器，所述存储器用于存储计算机程序代码，所述计算机程序代码包括计算机指令，当所述处理器执行所述计算机指令时，所述电子设备执行如权利要求1-7任意一项所述的三维模型检索方法。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质中存储有计算机程序，所述计算机程序包括程序指令，所述程序指令当被电子设备的处理器执行时，使所述处理器执行如权利要求1-7任意一项所述的三维模型检索方法。

技术总结
本发明公开了一种三维模型检索方法、装置及计算机可读存储介质。其中，方法包括：获取待进行模型检索的三维模型文件；基于预训练好的人工智能模型对三维模型文件进行特征提取，得到三维模型文件的特征向量；根据特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型。本发明通过预训练好的人工智能模型提取待进行模型检索的三维模型文件的特征向量，并根据特征向量，采用基于相似性搜索类库的特征检索引擎在三维模型数据库中进行模型检索，得到相似性最高的前预设数量个目标三维模型，实现了三维模型的检索，且提高了检索精度和检索效率。

技术研发人员：容汉铿,赖官壬,邵军,曾建力,张晴浩,李楠,谢巧琳
受保护的技术使用者：粤港澳大湾区（广东）国创中心
技术研发日：
技术公布日：2024/1/15