rpmcruz / pnets

Utilities for PointNet and similar networks

pnets

These are a set of evolving utilities for PyTorch to make it easy to do research on PointNet and similar networks. Not all code is mine; such code is acknowledged accordingly.

There are other interesting packages like Torch Points 3D, but we had some problems installing it and decided to build our own package for fun and profit. :-)

Structure

The package is made of the following modules:

• aug: augmentation methods
• data: datasets that we come with
• metrics: metrics for point-clouds
• plot: plotting routines
• pointnet: a vanilla PointNet implementation (for classification and segmentation)

You can have a look at some examples under the src folder.

Install

pip3 install git+https://github.com/rpmcruz/pnets.git