/LearningPatches

Code for Learning Manifold Patch-Based Representations of Man-Made Shapes, in ICLR 2021.

Primary LanguagePythonMIT LicenseMIT

LearningPatches | Webpage | Paper | Video

Learning Manifold Patch-Based Representations of Man-Made Shapes

Learning Manifold Patch-Based Representations of Man-Made Shapes
Dmitriy Smirnov, Mikhail Bessmeltsev, Justin Solomon
International Conference on Learning Representations (ICLR) 2021

Set-up

To install the code, run:

conda create -n learningpatches python=3.6 -y
conda activate learningpatches
conda install pytorch=1.3.1 torchvision cudatoolkit=10.2 -c pytorch -y
pip install -r requirements.txt

Also, be sure to execute export PYTHONPATH=:$PYTHONPATH prior to running any of the scripts.

Demo

First, download the pretrained models for each shape category:

wget -O models.zip https://www.dropbox.com/s/ntt1ytpjwx2385i/learningpatches_models.zip?dl=0
unzip models.zip

Then, run the following to generate an OBJ file with the 3D model for a given input sketch PNG image:

python scripts/run.py demo/airplane.png airplanes out.obj --no-turbines

Make sure to specify airplanes, bathtubs, bottles, cars, guitars, guns, knives, or guns as the shape category. Optionally, for the airplanes category, the --no-turbines flag does not output the turbine patches in the 3D model.

The demo directory contains PNGs of some sample input sketches.

Note that the meshes output by the demo script may have non-manifold discontinuities between patches due to discretization artifacts. This can be avoided by choosing the number of subdivisions based on patch boundary arc lengths. The results shown in the paper are all computed in this way.

BibTeX

@inproceedings{smirnov2021patches,
  title={Learning Manifold Patch-Based Representations of Man-Made Shapes},
  author={Smirnov, Dmitriy and Bessmeltsev, Mikhail and Solomon, Justin},
  year={2021},
  booktitle={International Conference on Learning Representations (ICLR)}
}