基于跨实例和类别对比的三维模型检索方法及系统

专利主权项内容

1.基于跨实例和类别对比的三维模型检索方法，其特征是，包括：获取查询图像、查询图像的掩码图以及三维模型集合，每个三维模型均设有三维模型的多视图；对查询图像进行图像增强得到若干个增强后的查询图像；将增强后的查询图像、查询图像的掩码图、三维模型的多视图均输入到训练后的三维模型检索网络中，输出三维模型检索结果；其中，训练后的三维模型检索网络，用于：将增强后的查询图像和查询图像的掩码图，均输入到第一卷积神经网络中，输出查询图像的特征表示；将三维模型的多视图输入到第二卷积神经网络中，输出三维模型每个视图的特征表示，将三维模型所有视图的特征表示进行加权求和，得到三维模型的特征表示；将查询图像的特征表示和三维模型的特征表示，均输入到分类器中，训练后的分类器输出查询图像所对应的三维模型；其中，训练后的三维模型检索网络，总损失函数为第一、第二和第三损失函数的求和结果；第一损失函数用于缩小查询图像与三维模型之间的距离；第二损失函数用于缩小增强后的查询图像与其他增强后的查询图像之间的距离；第三损失函数用于增大不同实例三维模型之间的距离。