Home

Awesome

Hierachical Surface Prediction

Installation

Install torch

Download CImg and place it in the torch-hsp subfolder. The file "CImg.h" needs to be in the path "torch-hsp/CImg/".

Install the torch package torch-hsp by running "luarocks make hsp-1.0-0.rockspec" in the torch-hsp folder

Running Demo

A demo script is included which reconstructs a single image and outputs a mesh as obj file. It needs as input the pretrained network provided here.

th hspDemo.lua <GPU ID> <Trained Network File Name> <Input Image File Name>

Training Network

Example parameter files are provided here.

The data is provided here.

To train a network the paths to the shapenet dataset and the output folder in the "parameters.lua" file need to be adjusted first.

th trainNetworkHierarchical.lua <GPU ID> <Parameter File Name>

License and Citation

The code is released as GPLv2.

When using the provided data make sure to respect the shapenet license.

Please cite our paper when using the code.<br /> C. Häne, S. Tulsiani, J. Malik, <B>Hierarchical Surface Prediction for 3D Object Reconstruction</B>, Proc. Int. Conf. on 3D Vision (3DV), 2017