Closed dkoh0207 closed 2 years ago
adapt_labels_knn will use torch_cluster.knn to assign predicted nonghost / true ghost voxels the nearest neighbor's label information. This saves CUDA memory and shortens iteration time while keeping the same behavior as that of adapt_labels.
adapt_labels_knn
torch_cluster.knn
adapt_labels
adapt_labels_knnwill usetorch_cluster.knnto assign predicted nonghost / true ghost voxels the nearest neighbor's label information. This saves CUDA memory and shortens iteration time while keeping the same behavior as that ofadapt_labels.