SGPN

by laughtervv

laughtervv / SGPN

SGPN:Similarity Group Proposal Network for 3D Point Cloud Instance Segmentation, CVPR, 2018

212 Stars 48 Forks Last release: Not found MIT License 3 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

SGPN:Similarity Group Proposal Network for 3D Point Cloud Instance Segmentation [Arxiv]

Dependencies

  • tensorflow
    (1.3.0)
  • h5py

Training & Testing

We firstly split the training set into training part and validation part. SGPN is finetuned on a pre-trained semantic segmentation model with large batchsize. For training,

bash
python train.py 
Use the following scripts to generate results.
valid.py
is used to compute the per-category theshold for group merging. We then use Scannet Evaluation to evaluate test results.
bash
python valid.py
python generate_results.py

Data and Model

Please refer to

data/
for example h5 file and input list file. A pre-trained model can be downloaded [here].

Citation

If you find our work useful, please consider citing:

    @inproceedings{wang2018sgpn,
        title={SGPN: Similarity Group Proposal Network for 3D Point Cloud Instance Segmentation},
        author={Wang, Weiyue and Yu, Ronald and Huang, Qiangui and Neumann, Ulrich},
        booktitle={CVPR},
        year={2018}
    }

Acknowledgemets

This project is built upon [PointNet] and [PointNet++].

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.