Skip to content

jnypan/TMNet

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

TMNet

This repository contains the source codes for the paper Deep Mesh Reconstruction from Single RGB Images via Topology Modification Networks.

Citing this work

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

@inproceedings{pan2019deep,
  title={Deep Mesh Reconstruction from Single RGB Images via Topology Modification Networks},
  author={Pan, Junyi and Han, Xiaoguang and Chen, Weikai and Tang, Jiapeng and Jia, Kui},
  booktitle={Proceedings of the IEEE International Conference on Computer Vision},
  pages={9964--9973},
  year={2019}
}

Install

Clone the repo and install dependencies

This implementation uses Pytorch 1.0.0.

## Download the repository
git clone https://github.com/jnypan/TMNet.git
cd TMNet
## Create python env with relevant packages
conda create --name tmnet python=3.7
source activate tmnet
pip install pandas visdom
cd ./extension
python setup.py install

Download

Data

We used the ShapeNet dataset for 3D models, and rendered views from 3D-R2N2:

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

Usage

Train

bash train/train.sh

Generate

python generate.py --model $model_path

Evaluation results

The evaluation is based on 2,048 ground truth points and 2,048 points uniformly sampled from the generated meshes. The CD is multiplied by 1,000 and the EMD is multiplied by 100. The threshold for F-score is set to be 0.002.

Chair CD EMD F-score
OccNet 6.647 13.266 69.08
AtlasNet 5.415 12.106 70.51
Ours 4.850 11.256 73.72

About

This repository contains the code for the paper "Deep Mesh Reconstruction from Single RGB Images ".

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages