Skip to content

Point Cloud Upsampling via Disentangled Refinement, CVPR, 2021

License

Notifications You must be signed in to change notification settings

liruihui/Dis-PU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Point Cloud Upsampling via Disentangled Refinement

This repository contains a Tensorflow implementation of the paper:

Point Cloud Upsampling via Disentangled Refinement.
Ruihui Li, Xianzhi Li, Pheng-Ann Heng, Chi-Wing Fu.

CVPR 2021

Getting Started

  1. Clone the repository:

    https://github.com/liruihui/Dis-PU.git
    cd Dis-PU

    Installation instructions for Ubuntu 16.04:

  2. Compile the customized TF operators by sh complile_op.sh. Follow the information from here to compile the TF operators.

  3. Train the model: First, you need to download the training patches in HDF5 format from GoogleDrive and put it in folder data. Then run:

    cd code
    python dis-pu.py --phase train
  4. Evaluate the model: First, you need to download the pretrained model from GoogleDrive, extract it and put it in folder 'model'. Then run:

    cd code
    python dis-pu.py --phase test

    You will see the input and output results in the folder data/test/output.

  5. The training and testing mesh files can be downloaded from GoogleDrive.

Evaluation code

We provide the evaluation code. In order to use it, you need to install the CGAL library. Please refer this link and PU-Net to install this library. Then:

cd evaluation_code
cmake .
make
./evaluation Icosahedron.off Icosahedron.xyz

The second argument is the mesh, and the third one is the predicted points.

Citation

If Dis-PU is useful for your research, please consider citing:

@inproceedings{li2021dispu,
     title={Point Cloud Upsampling via Disentangled Refinement},
     author={Li, Ruihui and Li, Xianzhi and Heng, Pheng-Ann and Fu, Chi-Wing},
     booktitle = {Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
     year = {2021}
 }

Questions

Please contact 'lirh@cse.cuhk.edu.hk'

About

Point Cloud Upsampling via Disentangled Refinement, CVPR, 2021

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published