GSPN: Generative Shape Proposal Network for 3D Instance Segmentation in Point Cloud
About
We introduce a novel 3D object proposal approach named Generative Shape Proposal Network (GSPN) for instance segmentation in point cloud data. Instead of treating object proposal as a direct bounding box regression problem, we take an analysis-by-synthesis strategy and generate proposals by reconstructing shapes from noisy observations in a scene. We incorporate GSPN into a novel 3D instance segmentation framework named Region-based PointNet (R-PointNet) which allows flexible proposal refinement and instance segmentation generation. We achieve state-of-the-art performance on several 3D instance segmentation tasks. The success of GSPN largely comes from its emphasis on geometric understandings during object proposal, which greatly reducing proposals with low objectness.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Object Detection | ScanNet V2 (val) | mAP@0.2562.8 | 352 | |
| 3D Instance Segmentation | ScanNet V2 (val) | Average AP5037.8 | 195 | |
| 3D Instance Segmentation | ScanNet v2 (test) | mAP30.6 | 135 | |
| 3D Instance Segmentation | S3DIS (Area 5) | -- | 106 | |
| 3D Instance Segmentation | S3DIS (6-fold CV) | Mean Precision @50% IoU38.2 | 92 | |
| 3D Instance Segmentation | ScanNet hidden v2 (test) | Cabinet AP@0.534.8 | 69 | |
| 3D Object Detection | ScanNet (val) | mAP@0.2530.6 | 66 | |
| Instance Segmentation | ScanNetV2 (val) | mAP@0.523.5 | 58 | |
| 3D Object Detection | ScanNet V2 | AP5017.7 | 54 | |
| Instance Segmentation | PartNet 1.0 (test) | mAP (Chair)26.8 | 44 |