Table of contents

1. Installation
2. Data Preparation
3. Training and Testing
4. Quick Demo
5. Acknowledgments
6. Contacts

GaPro: Box-Supervised 3D Point Cloud Instance Segmentation Using Gaussian Processes as Pseudo Labelers

Tuan Duc Ngo, Binh-Son Hua, Khoi Nguyen
VinAI Research, Vietnam

Abstract: Instance segmentation on 3D point clouds (3DIS) is a longstanding challenge in computer vision, where state-of-the-art methods are mainly based on full supervision. As annotating ground truth dense instance masks is tedious and expensive, solving 3DIS with weak supervision has become more practical. In this paper, we propose GaPro, a new instance segmentation for 3D point clouds using axis-aligned 3D bounding box supervision. Our two-step approach involves generating pseudo labels from box annotations and training a 3DIS network with the resulting labels. Additionally, we employ the self-training strategy to improve the performance of our method further. We devise an effective Gaussian Process to generate pseudo instance masks from the bounding boxes and resolve ambiguities when they overlap, resulting in pseudo instance masks with their uncertainty values. Our experiments show that GaPro outperforms previous weakly supervised 3D instance segmentation methods and has competitive performance compared to state-of-the-art fully supervised ones. Furthermore, we demonstrate the robustness of our approach, where we can adapt various state-of-the-art fully supervised methods to the weak supervision task by using our pseudo labels for training.

Details of the model architecture and experimental results can be found in our paper:

@inproceedings{ngo2023gapro,
 author={Tuan Duc Ngo, Binh-Son Hua, Khoi Nguyen},
 booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
 title={GaPro: Box-Supervised 3D Point Cloud Instance Segmentation \\Using Gaussian Processes as Pseudo Labelers},
 year= {2023}
}

Please CITE our paper whenever this repository is used to help produce published results or incorporated into other software.

Installation 📝

Please refer to installation guide.

Data Preparation 🔨

Please refer to data preparation.

Training and Testing 🚆

Please refer to training guide.

Quick Demo 🔥

ScanNetv2

Method	Dataset	AP	AP_50	Config	Checkpoint
GaPro + ISBNet	Test	49.3	69.8
''	Val	50.6	69.1	config	Google Drive
GaPro + SPFormer	Test	48.2	68.2
''	Val	51.1	70.4	config	Google Drive

S3DIS: TBD

Method	Dataset	AP	AP_50	Config	Checkpoint
GaPro + ISBNet	Area 5	56.3	67.5	TBD	TBD

Run evaluation with pre-trained models:

cd ISBNet/ # or SPFormer
python3 tools/test.py <path_to_config_file> <path_to_pretrain_weight>

Acknowledgements 👏

This repo is built upon SpConv, ISBNet, and SPFormer.

Contacts 📧

If you have any questions or suggestions about this repo, please feel free to contact me ([email protected]).