torch-points-kernels

PyTorch kernels for spatial operations on point clouds


License
MIT
Install
pip install torch-points-kernels==0.6.10

Documentation

3D Point Cloud Kernels

Pytorch CPU and CUDA kernels for spatial search and interpolation for 3D point clouds.

Deploy Unittests

Installation

Requires torch version 1.0 or higher. To install a specific version replace the version number with a given tag.

pip install git+https://github.com/nicolas-chaulet/torch-points-kernels.git#v0.5.0

or with poetry:

poetry add git+https://github.com/nicolas-chaulet/torch-points.git#v0.5.0

Usage

import torch
import torch_points_kernels.points_cuda

Build and test

python setup.py build_ext --inplace
python -m unittest

Projects using those kernels.

Pytorch Point Cloud Benchmark

Credit