SoftGroup for 3D Instance Segmentation on Point Clouds

By Thang Vu, Kookhoi Kim, Tung Minh Luu, Xuan Thanh Nguyen, Chang-Dong Yoo. Computer Vision and Pattern Recognition, 2022

350 citations recorded by the source corpus.

Existing state-of-the-art 3D instance segmentation methods perform semantic segmentation followed by grouping. The hard predictions are made when performing semantic segmentation such that each point is associated with a single class. However, the errors stemming from hard decision propagate into grouping that results in (1) low overlaps between the predicted instance with the ground truth and (2) substantial false positives. To address the aforementioned problems, this paper proposes a 3D instance segmentation method referred to as SoftGroup by performing bottom-up soft grouping followed by top-down refinement. SoftGroup allows each point to be associated with multiple classes to mitigate the problems stemming from semantic prediction errors and suppresses false positive instances by learning to categorize them as background. Experimental results on different datasets and multiple evaluation metrics demonstrate the efficacy of SoftGroup. Its performance surpasses the strongest prior method by a significant margin of $+6.2\%$ on the ScanNet v2 hidden test set and $+6.8\%$ on S3DIS Area 5 in terms of $AP_{50}$. Soft-Group is also fast, running at 345ms per scan with a sin-gle Titan X on ScanNet v2 dataset. The source code and trained models for both datasets are available at https://github.com/thangvubk/SoftGroup.git.

Subfields: Aerospace Engineering, Computational Mechanics, Geology

View Paper: https://www.semanticscholar.org/paper/a35420f502f1104db871c946db1e2ddf44dfe58c

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Aerospace Engineering

  1. Matthias Nießner
  2. Angela Dai
  3. Marc Pollefeys
  4. Jiajun Wu
  5. Silvio Savarese
  6. Francis Engelmann
  7. Roberto Mart'in-Mart'in
  8. Hongsheng Li

In Computational Mechanics

  1. Matthias Nießner
  2. Angela Dai
  3. Jiajun Wu
  4. Silvio Savarese
  5. Marc Pollefeys
  6. P. Heng
  7. Francis Engelmann
  8. Chandan Yeshwanth

In Geology

  1. Matthias Nießner
  2. Angela Dai
  3. Marc Pollefeys
  4. Silvio Savarese
  5. Francis Engelmann
  6. Jiajun Wu
  7. Yueh-Cheng Liu
  8. Chandan Yeshwanth

Citations in the corpus