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. :-)
The package is made of the following modules:
aug
: augmentation methodsdata
: datasets that we come withmetrics
: metrics for point-cloudsplot
: plotting routinespointnet
: a vanilla PointNet implementation (for classification and segmentation)
You can have a look at some examples under the src
folder.
pip3 install git+https://github.com/rpmcruz/pnets.git