Pointnet2 PyTorch

Partial implemention of Pointnet2 written in PyTorch.

The custom ops used by Pointnet2 are currently ONLY supported on the GPU using CUDA.

Building CUDA kernels

• mkdir build && cd build
• cmake .. && make

Exampling training

Two training examples are provided by train_sem_seg.py and train_cls.py. The datasets for both will be downloaded automatically by default