Skip to content

[CVPR 2022, Oral] Learning Partial point cloud matching in Rigid and Deformable scenes

License

Notifications You must be signed in to change notification settings

rabbityl/lepard

Folders and files

NameName
Last commit message
Last commit date

Latest commit

23a93d0 · Mar 24, 2023

History

17 Commits
Mar 7, 2022
Nov 24, 2021
Nov 24, 2021
Nov 24, 2021
Nov 24, 2021
Dec 24, 2021
Nov 24, 2021
Nov 24, 2021
Mar 24, 2023
Mar 24, 2022
Nov 24, 2021
Nov 24, 2021

Repository files navigation

Lepard: Learning Partial point cloud matching in Rigid and Deformable scenes [Paper]

drawing

Method overview

drawing

4DMatch Benchmark

4DMatch is a benchmark for matching and registration of partial point clouds with time-varying geometry. It is constructed using randomly selected 1761 sequences from DeformingThings4D. Below shows point cloud pairs with different overlap ratios.

drawing

Installation

We tested the code on python 3.8.10; Pytroch version '1.7.1' or '1.9.0+cu111'; GPU model GeForce RTX-2080 or Nvidia A100.

conda env create -f environment.yml
conda activate lepard
cd cpp_wrappers; sh compile_wrappers.sh; cd ..

Download data and pretrained model

Train and evaluation on 4DMatch

Download and extract the 4DMatch split to your custom folder. Then update the data_root in configs/train/4dmatch.yaml and configs/test/4dmatch.yaml

  • Evaluate pre-trained
python main.py configs/test/4dmatch.yaml

(To switch between 4DMatch and 4DLoMatch benchmark, modify the split configuration in configs/test/4dmatch.yaml)

  • Train from scratch
python main.py configs/train/4dmatch.yaml

Intergration to Non-rigid Registration

An exmaple can be found here: Nonrigid-ICP-Pytorch

Train and evaluation on 3DMatch

Download and extract the 3DMatch split to your custom folder. Then update the data_root in configs/train/3dmatch.yaml and configs/test/3dmatch.yaml

  • Evaluate pre-trained
python main.py configs/test/3dmatch.yaml

(To switch between 3DMatch and 3DLoMatch benchmark, modify the split configuration in configs/test/3dmatch.yaml)

  • Train from scratch
python main.py configs/train/3dmatch.yaml

Citation

If you use Lepard code or 4DMatch data please cite:

@article{lepard2021, 
    title={Lepard: Learning partial point cloud matching in rigid and deformable scenes.}, 
    author={Yang Li and Tatsuya Harada},
    journal={IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    year={2022}
}

About

[CVPR 2022, Oral] Learning Partial point cloud matching in Rigid and Deformable scenes

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages