Awesome

SC^2-PCR: A Second Order Spatial Compatibility for Efficient and Robust Point Cloud Registration (CVPR 2022)

PyTorch implementation of the paper:

SC^2-PCR: A Second Order Spatial Compatibility for Efficient and Robust Point Cloud Registration.

Zhi Chen, Kun Sun, Fan Yang, Wenbing Tao.

Introduction

In this paper, we present a second order spatial compatibility (SC^2) measure based method for efficient and robust point cloud registration (PCR), called SC^2-PCR. Firstly, we propose a second order spatial compatibility (SC^2) measure to compute the similarity between correspondences. It considers the global compatibility instead of local consistency, allowing for more distinctive clustering between inliers and outliers at early stage. Based on this measure, our registration pipeline employs a global spectral technique to find some reliable seeds from the initial correspondences. Then we design a two-stage strategy to expand each seed to a consensus set based on the SC^2 measure matrix. Finally, we feed each consensus set to a weighted SVD algorithm to generate a candidate rigid transformation and select the best model as the final result. Our method can guarantee to find a certain number of outlier-free consensus sets using fewer samplings, making the model estimation more efficient and robust. In addition, the proposed SC^2 measure is general and can be easily plugged into deep learning based frameworks. Extensive experiments are carried out to investigate the performance of our method.

Requirements

If you are using conda, you may configure SC2-PCR as:

conda env create -f environment.yml
conda activate SC2_PCR

3DMatch

Data preparation

Downsample and extract FPFH and FCGF descriptors for each frame of the 3DMatch test dataset. Here we provide the processed test set with pre-computed FPFH/FCGF descriptors. The data should be organized as follows:

--data--3DMatch                
        ├── fragments                 
        │   ├── 7-scene-redkitechen/
        |   |   ├── cloud_bin_0.ply
        |   |   ├── cloud_bin_0_fcgf.npz
        |   |   ├── cloud_bin_0_fpfh.npz
        │   |   └── ...      
        │   ├── sun3d-home_at-home_at_scan1_2013_jan_1/      
        │   └── ...                
        ├── gt_result                   
        │   ├── 7-scene-redkitechen-evaluation/   
        |   |   ├── 3dmatch.log
        |   |   ├── gt.info
        |   |   ├── gt.log
        │   |   └── ...
        │   ├── sun3d-home_at-home_at_scan1_2013_jan_1-evaluation/
        │   └── ...                               

Testing

Use the following command for testing.

python ./test_3DMatch.py --config_path config_json/config_3DMatch.json

The CUDA_DEVICE and basic parameters can be changed in the json file.

3DLoMatch

Data preparation

FPFH and FCGF descriptors can be prepared in the same way as testing 3DMatch. If you want to test the predator descriptor, you should first follow the offical instruction of predator to extract the descriptors for 3DMatch dataset and organize the data as follows:

--data--3DLoMatch                
        ├── 0.pth        
        ├── 1.pth                 
        ├── ...  
        └── 1780.pth

Testing

Use the following command for testing.

python ./test_3DLoMatch.py --config_path config_json/config_3DLoMatch.json

KITTI odometry

Data preparation

Downsample and extract FPFH and FCGF descriptors for each frame of the KITTI test dataset. The raw point clouds can be download from KITTI Odometry website.. For your convenience, here we provide the pre-computed FPFH and FCGF descriptors for the KITTI test set.

--data--KITTI                
        ├── fpfh_test                 
        │   ├── pair_0.npz        
        |   ├── pair_1.npz                
        |   ├── ...  
        |   └── pair_554.npz
        ├── fcgf_test                
        │   ├── pair_0.npz        
        |   ├── pair_1.npz                
        |   ├── ...  
        |   └── pair_554.npz

Testing

Use the following command for testing.

python ./test_KITTI.py --config_path config_json/config_KITTI.json

Results

3DMatch

We evaluate SC^2-PCR on the standard 3DMatch benchmarks:

3DMatch

We evaluate SC^2-PCR on the standard 3DLoMatch benchmarks:

KITTI odometry

We evaluate SC^2-PCR on the standard KITTI benchmarks:

Citation

@InProceedings{Chen_2022_CVPR,
    author    = {Chen, Zhi and Sun, Kun and Yang, Fan and Tao, Wenbing},
    title     = {SC2-PCR: A Second Order Spatial Compatibility for Efficient and Robust Point Cloud Registration},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2022},
    pages     = {13221-13231}
}

Acknowledgements

• PointDSC
• FCGF
• DGR
• PREDATOR