/IGR

Implicit Geometric Regularization for Learning Shapes

Primary LanguagePython

IGR: Implicit Geometric Regularization for Learning Shapes

This repository contains an implementation to the ICML 2020 paper: "Implicit Geometric Regualrization for Learning Shapes".

IGR is a deep learning approach for learning implicit signed distance representations directly from raw point clouds with or without normal data. Our method aims to find an SDF by optimizing the network to solve the eikonal equation with the input point cloud as boundary condition. Although this is an ill posed condition we enjoy an implicit regualrization coming from the optimization procedure itself which aims our method to simple natrual solutions as can be seen on the figure above.

For more details:

paper: https://arxiv.org/abs/2002.10099.

video: https://youtu.be/6cOvBGBQF9g.

Installation Requirmenets

The code is compatible with python 3.7 and pytorch 1.2. In addition, the following packages are required:
numpy, pyhocon, plotly, scikit-image, trimesh.

Usage

Surface reconstruction

IGR can be used to reconstruct a single surface given a point cloud with or without normal data. Adjust reconstruction/setup.json to the path of the input 2D/3D point cloud:

train
{
  ...
  d_in=D
  ...
  input_path = your_path
  ...
}

Where D=3 in case we use 3D data or 2 if we use 2D. We support xyz,npy,npz,ply files.

Then, run training:

cd ./code
python reconstruction/run.py 

Finally, to produce the meshed surface, run:

cd ./code
python reconstruction/run.py --eval --checkpoint CHECKPOINT

where CHECKPOINT is the epoch you wish to evaluate of 'latest' if you wish to take the most recent epoch.

Learning shapespace from the D-Faust oriented point clouds

Data

The raw scans can be downloaded from http://dfaust.is.tue.mpg.de/downloads. In order to sample point clouds with normals use:

cd ./code
python preprocess/dfaust.py --src-path SRC_PATH --out-path OUT_PATH

where SRC_PATH is the absoule path of the directory with the original D-Faust scans, and OUT_PATH is the absolute path of the directory on which you wish to output the processed point clouds.

It is also possible to process only train data with input --mode 0 our only test with --mode 1.

In case you wish to only process part of the data (e.g. for parallel processing) it is possible by adding --names NAME_1,NAME_2,...,NAME_k where NAME_i is one of D-Faust shapes e.g. 50002, 50020.

After preprocessing ended adjust the file ./shapespace/dfaust_setup.conf to the cur path of the data:

train
{
  ...
  dataset_path = OUT_PATH/dfaust_processed
  ...
}

Predicting meshed surfaces with IGR pretrained network

We have uploaded IGR trained network. To produce predictions on unseen test scans, run:

cd ./code
python shapespace/eval.py --checkpoint 1200 --exp-name dfaust_pretrained --split dfaust/test_all.json --exps-dir trained_models

In case you wish to generate less models you can use --split dfaust/test_models.json

Interpolating latents of IGR pretrained network

To meshs of latent interpolation between two shapes use:

cd ./code
python shapespace/interpolate.py --interval INTERVAL --checkpoint 1200 --exp-name dfaust_pretrained --exps-dir trained_models

Where INTERVAL is the number (int) linspace of latent interpolations.

In case you wish to interpolate different shapes adjust the file dfaust/interpolation.json

Training

If you want to train IGR yourself, run:

cd ./code
python shapespace/train.py

Citation

If you find our work useful in your research, please consider citing:

@incollection{icml2020_2086,
 author = {Gropp, Amos and Yariv, Lior and Haim, Niv and Atzmon, Matan and Lipman, Yaron},
 booktitle = {Proceedings of Machine Learning and Systems 2020},
 pages = {3569--3579},
 title = {Implicit Geometric Regularization for Learning Shapes},
 year = {2020}
}

Related papers