• Stars
    star
    119
  • Rank 297,930 (Top 6 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created over 3 years ago
  • Updated almost 3 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

MVP Benchmark for Multi-View Partial Point Cloud Completion and Registration

MVP Benchmark: Multi-View Partial Point Clouds for Completion and Registration

[Website]

[NEWS]

[Timeline]

  • 2021-07-12   Submission start date
  • 2021-09-12   Public submission deadline
  • 2021-09-19   Private submission deadline
  • 2021-10-04   Technical report deadline
  • 2021-10-17   Awards at ICCV2021 Workshop

[MVP Benchmark]

Overview

This repository introduces the MVP Benchmark for partial point cloud COMPLETION and REGISTRATION, and it also includes following recent methods:

This repository is implemented in Python 3.7, PyTorch 1.5.0, CUDA 10.1 and gcc > 5.

Installation

Install Anaconda, and then use the following command:

git clone --depth=1 https://github.com/paul007pl/MVP_Benchmark.git
cd MVP_Benchmark; source setup.sh;

If your connection to conda and pip is unstable, it is recommended to manually follow the setup steps in setup.sh.

MVP Dataset

Download corresponding dataset:

Usage

For both completion and registration:

  • cd completion or cd registration

  • To train a model: run python train.py -c ./cfgs/*.yaml, e.g. python train.py -c ./cfgs/pcn.yaml

  • To test a model: run python test.py -c ./cfgs/*.yaml, e.g. python test.py -c ./cfgs/pcn.yaml

  • Config for each algorithm can be found in cfgs/.

  • run_train.sh and run_test.sh are provided for SLURM users.

  • Different partial point clouds for the same CAD Model:

  • High-quality complete point clouds:


[Citation]

If you find our code useful, please cite our paper:

@inproceedings{pan2021variational,
  title={Variational Relational Point Completion Network},
  author={Pan, Liang and Chen, Xinyi and Cai, Zhongang and Zhang, Junzhe and Zhao, Haiyu and Yi, Shuai and Liu, Ziwei},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={8524--8533},
  year={2021}
}
@article{pan2021robust,
  title={Robust Partial-to-Partial Point Cloud Registration in a Full Range},
  author={Pan, Liang and Cai, Zhongang and Liu, Ziwei},
  journal={arXiv preprint arXiv:2111.15606},
  year={2021}
}
@article{pan2021mvp, 
  title={Multi-View Partial (MVP) Point Cloud Challenge 2021 on Completion and Registration: Methods and Results}, 
  author={Pan, Liang and Wu, Tong and Cai, Zhongang and Liu, Ziwei and Yu, Xumin and Rao, Yongming and Lu, Jiwen and Zhou, Jie and Xu, Mingye and Luo, Xiaoyuan and Fu, Kexue, and Gao, Peng, and Wang, Manning, and Wang, Yali, and Qiao, Yu, and Zhou, Junsheng, and Wen, Xin, and Xiang, Peng, and Liu, Yu-Shen, and Han, Zhizhong, and Yan, Yuanjie, and An, Junyi, and Zhu, Lifa, and Lin, Changwei, and Liu, Dongrui, and Li, Xin, and G ́omez-Fern ́andez, Francisco, and Wang, Qinlong, and Yang, Yang}, 
  journal={arXiv preprint arXiv:2112.12053},
  year={2021}
}

[License]

Our code is released under Apache-2.0 License.


[Acknowledgement]

We include the following PyTorch 3rd-party libraries:
[1] CD
[2] EMD
[3] MMDetection3D

We include the following algorithms:
[1] PCN
[2] ECG
[3] VRCNet
[4] DCP
[5] DeepGMR
[6] IDAM