公开(公告)号：US11869149B2

公开(公告)日：2024-01-09

申请号：US17744467

申请日：2022-05-13

Applicant: NVIDIA CORPORATION

Inventor： Ben Eckart ,  Christopher Choy ,  Chao Liu ,  Yurong You

IPC: G06T17/10 ,  G06T19/20 ,  G06N3/084 ,  G06N3/045

CPC classification number: G06T17/10 ,  G06N3/045 ,  G06N3/084 ,  G06T19/20 ,  G06T2219/2016

Abstract: In various embodiments, an unsupervised training application executes a neural network on a first point cloud to generate keys and values. The unsupervised training application generates output vectors based on a first query set, the keys, and the values and then computes spatial features based on the output vectors. The unsupervised training application computes quantized context features based on the output vectors and a first set of codes representing a first set of 3D geometry blocks. The unsupervised training application modifies the first neural network based on a likelihood of reconstructing the first point cloud, the quantized context features, and the spatial features to generate an updated neural network. A trained machine learning model includes the updated neural network, a second query set, and a second set of codes representing a second set of 3D geometry blocks and maps a point cloud to a representation of 3D geometry instances.