-
公开(公告)号:CN113961737A
公开(公告)日:2022-01-21
申请号:CN202111128474.0
申请日:2021-09-26
Applicant: 中国科学院计算技术研究所
IPC: G06F16/583 , G06F16/532 , G06N3/08
Abstract: 本发明提出一种三维模型检索模型训练方法,包括以下步骤:将三维模型数据库中的三维模型进行渲染以获得对应的多视角渲染灰度图像;在训练的查询图像小批次中,将该查询图像进行颜色转换以获得对应的数据增强图像;将该数据增强图像与其对应的掩码图像送入查询图像编码器中以获得该查询图像对应的查询编码;将该三维模型对应的多视角渲染灰度图像送入渲染编码器中以获得该三维模型对应的多视角渲染灰度图像对应的渲染编码;将该查询编码与该渲染编码送入注意力机制模块中以获得该三维模型针对于当前查询图像的特定编码;利用损失函数进行优化以获得该三维模型检索模型。