Top-Down Beats Bottom-Up in 3D Instance Segmentation

6 Feb 2023  ·  Maksim Kolodiazhnyi, Anna Vorontsova, Anton Konushin, Danila Rukhovich ·

Most 3D instance segmentation methods exploit a bottom-up strategy, typically including resource-exhaustive post-processing. For point grouping, bottom-up methods rely on prior assumptions about the objects in the form of hyperparameters, which are domain-specific and need to be carefully tuned. On the contrary, we address 3D instance segmentation with a TD3D: the pioneering cluster-free, fully-convolutional and entirely data-driven approach trained in an end-to-end manner. This is the first top-down method outperforming bottom-up approaches in 3D domain. With its straightforward pipeline, it demonstrates outstanding accuracy and generalization ability on the standard indoor benchmarks: ScanNet v2, its extension ScanNet200, and S3DIS, as well as on the aerial STPLS3D dataset. Besides, our method is much faster on inference than the current state-of-the-art grouping-based approaches: our flagship modification is 1.9x faster than the most accurate bottom-up method, while being more accurate, and our faster modification shows state-of-the-art accuracy running at 2.6x speed. Code is available at https://github.com/SamsungLabs/td3d .

PDF Abstract

Results from the Paper


Ranked #5 on 3D Instance Segmentation on S3DIS (using extra training data)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
3D Instance Segmentation S3DIS TD3D AP@50 70.4 # 5
mAP 58.1 # 5
3D Instance Segmentation ScanNet(v2) TD3D mAP 48.9 # 10
mAP @ 50 75.1 # 7
mAP@25 87.5 # 2

Methods


No methods listed for this paper. Add relevant methods here