检索模型训练方法与装置、检索方法与装置、电子设备及存储介质

    公开(公告)号:CN113961737A

    公开(公告)日:2022-01-21

    申请号:CN202111128474.0

    申请日:2021-09-26

    Inventor: 高林 林明仙 杨洁

    Abstract: 本发明提出一种三维模型检索模型训练方法,包括以下步骤:将三维模型数据库中的三维模型进行渲染以获得对应的多视角渲染灰度图像;在训练的查询图像小批次中,将该查询图像进行颜色转换以获得对应的数据增强图像;将该数据增强图像与其对应的掩码图像送入查询图像编码器中以获得该查询图像对应的查询编码;将该三维模型对应的多视角渲染灰度图像送入渲染编码器中以获得该三维模型对应的多视角渲染灰度图像对应的渲染编码;将该查询编码与该渲染编码送入注意力机制模块中以获得该三维模型针对于当前查询图像的特定编码;利用损失函数进行优化以获得该三维模型检索模型。

Patent Agency Ranking