PointGroup-PyTorch

Reimplementation of PointGroup with different backbone(KPConv, DGCNN) for study purposes.

Reimplementation Process Post

Model Architecture

I used KPConv for Backbone, DGCNN for ScoreNet

Inference Result

Semantic segmentation

Instance segmentation

Training

• Data Preparation

Simply run the following script to start the training:

    python train.py

Testing

Simply run the following script to start the testing(recommended to run only 1 epoch):

    python test.py   # you should change weight directory

Reference

1. KPConv-PyTorch
2. dgcnn.pytorch
3. visualizing with pptk