Code base of ParSeNet: ECCV 2020.
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.
To install conda environment:
conda env create --force environment.yml -n parsenet
source activate parsenet
To dowload the dataset, run:
bash download_dataset.sh
For data organization, please see readme_data.md
.
Experiments are done on Nvidia 1080ti gpus.
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
python train_parsenet.py configs/config_parsenet.yml
python train_parsenet.py configs/config_parsenet_normals.yml
configs/config_parsenet_e2e.yml
(with 2 gpus). Further note that, this part of the training requires dynamic amount of gpu memory because a shape can have variable number of segment and corresponding number of fitting module. Training is done using Nvidia m40 (24 Gb gpu).python train_parsenet_e2e.py configs/config_parsenet_e2e.yml
test.py
python test.py 0 3998
@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}
}