Home

Awesome

ParSeNet: A Parametric Surface Fitting Network for 3D Point Clouds

Authors: Gopal Sharma, Difan Liu, Evangelos Kalogerakis, Subhransu Maji, Siddhartha Chaudhuri, Radomír Měch

This repository contains codebase for the ParSeNet paper published at ECCV-2020.

Paper | Project Page

Installation

To install conda environment:

conda env create --force environment.yml -n parsenet
source activate parsenet

Dataset

To dowload the dataset, run:

bash download_dataset.sh

For data organization, please see readme_data.md.


Experiments

Experiments are done on Nvidia 1080ti gpus.

SplineNet

python train_open_splines.py configs/config_open_splines.yml
python test_open_splines.py configs/config_test_open_splines.yml
python train_closed_control_points.py configs/config_closed_splines.yml
python test_closed_control_points.py configs/config_test_closed_splines.yml

ParSeNet

python train_parsenet.py configs/config_parsenet.yml
python train_parsenet.py configs/config_parsenet_normals.yml
python train_parsenet_e2e.py  configs/config_parsenet_e2e.yml
python test.py 0 3998

Acknowledgements

  1. This project takes inspiration of designing network architecture from the code base provided by Wang et.al.: https://github.com/WangYueFt/dgcnn
  2. We also thank Sebastian for timely release and advice on ABC dataset: https://deep-geometry.github.io/abc-dataset/

Citation

@misc{sharma2020parsenet,
    title={ParSeNet: A Parametric Surface Fitting Network for 3D Point Clouds},
    author={Gopal Sharma and Difan Liu and Evangelos Kalogerakis and Subhransu Maji and Siddhartha Chaudhuri and Radomír Měch},
    year={2020},
    eprint={2003.12181},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}