Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

FreePoint: Unsupervised Point Cloud Instance Segmentation

About

Instance segmentation of point clouds is a crucial task in 3D field with numerous applications that involve localizing and segmenting objects in a scene. However, achieving satisfactory results requires a large number of manual annotations, which is a time-consuming and expensive process. To alleviate dependency on annotations, we propose a novel framework, FreePoint, for underexplored unsupervised class-agnostic instance segmentation on point clouds. In detail, we represent the point features by combining coordinates, colors, and self-supervised deep features. Based on the point features, we perform a bottom-up multicut algorithm to segment point clouds into coarse instance masks as pseudo labels, which are used to train a point cloud instance segmentation model. We propose an id-as-feature strategy at this stage to alleviate the randomness of the multicut algorithm and improve the pseudo labels' quality. During training, we propose a weakly-supervised two-step training strategy and corresponding losses to overcome the inaccuracy of coarse masks. FreePoint has achieved breakthroughs in unsupervised class-agnostic instance segmentation on point clouds and outperformed previous traditional methods by over 18.2% and a competitive concurrent work UnScene3D by 5.5% in AP. Additionally, when used as a pretext task and fine-tuned on S3DIS, FreePoint performs significantly better than existing self-supervised pre-training methods with limited annotations and surpasses CSC by 6.0% in AP with 10% annotation masks.

Zhikai Zhang, Jian Ding, Li Jiang, Dengxin Dai, Gui-Song Xia• 2023

Related benchmarks

TaskDatasetResultRank
Instance SegmentationScanNet (val)
mAP18.9
39
Semantic Instance SegmentationS3DIS 10% masks
AP42.9
12
Semantic Instance SegmentationS3DIS 20% masks
AP48.1
12
Instance SegmentationScanNet (train)
AP21.4
6
Showing 4 of 4 rows

Other info

Code

Follow for update