一种三维CAD模型检索方法及装置

本申请提供一种三维CAD模型检索方法及装置，首先获取待检索三维CAD模型的二维图像集；计算二维图像集中图像的像素能量值，基于像素能量值排序将预设数量的二维图像作为有效视角图像；提取有效视角图像的特征嵌入向量；判断执行次数，当执行次数达到预设次数后，执行：聚合预设次数不同视角下的特征嵌入向量，得到嵌入向量集；将特征嵌入向量集作为待检索三维CAD模型的全局特征向量集；计算待检索三维CAD模型的全局特征向量集与三维CAD模型向量库中存储的向量的余弦相似度值，并根据余弦相似度值排序进行目标检索。分类任务计算全局特征向量与其分类标签之间的交叉熵，以增强预设数量特征向量之间的关联性。