ml-lab / 3dn

3DN: 3D Deformation Network, CVPR 2019

3DN: 3D Deformation Network [Arxiv]

Installation

Install Tensorflow (1.10.0), PyMesh.

The mesh sampling and mesh laplacian operations are under folder models/tf_ops. To build them, simply use cd models/tf_ops and sh compile.sh to compile. You may need to set CUDA path in each subfolder.

Training

cd shapenet/3D
python train.py

cd shapenet/2D
python train_vgg.py

Pretrained Model can be found here.

Testing

cd shapenet/3D
python test.py

cd shapenet/2D
python test.py

Citation

If you find this work useful, please consider citing:

@inproceedings{wang20193dn,
    title={3DN: 3D Deformation Network},
    author={Wang, Weiyue and Ceylan, Duygu and Mech, Radomir and Neumann, Ulrich},
    booktitle={CVPR},
    year={2019}
}

Acknowledgemets

• PointNet
• PointNet AutoEncoder
• Point Set Generation