Usage

The package was designed to be easily used through the experiments.

Here is an example of how you can import it

import numpy as np
import torch

from altametris.pctools.nearest_neighbors import nn_cpu, nn_gpu
from altametris.pctools.grid_subsampling import cpu as gs

# create a random point cloud
pc = np.random.rand(1000, 3)

# compute knn on CPU
idx = nn_cpu.knn(pc, pc, K=2)

# compute knn on GPU
pc_tensor = torch.from_numpy(pc)
idx = nn_gpu.knn(pc, pc, K=2)

# subsample
subpoints, _, _ = gs.subsample(points, grid_size=0.01)