Skip to content

wkqscut/DCGNet

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
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Deep Cascade Generation on Point Sets

Kaiqi Wang, Ke Chen*, Kui Jia     IJCAI 2019

[paper] | [project page]

This implementation uses Pytorch.

Installation

git clone https://github.com/wkqscut/DCGNet.git
cd DCGNet
## Create python env with relevant packages
conda create --name dcg python=3.7
conda activate dcg
pip install -U pip
pip install -r requirements.txt
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch  # cudatoolkit=10.0 for cuda10

Tested on pytorch >= 1.0 and python3.

Build

## Build chamfer distance
conda activate dcg
cd ./extension
python setup.py install
cd ../

Download

Dataset

We used the rendered imaged from 3d-R2N2, and the groundtruth 3D point clouds sampled from ShapeNet.

Pretrained models

## unzip the Pretrained models using the scripts
bash ./trained_models/unzip_models_dataset.sh

Run code

Demo

  • demo code for DCGNet
bash ./scripts/demo.sh

Training

Make sure that the visdom is alive before training:

python -m visdom.server -p 8990 (change the port if in use)
  • train the DCGNet for Point Set AutoEncoding:
bash ./scripts/train_svr_dcg.sh
  • train the DCGNet for Point Set Reconstruction from a Single Image:
bash ./scripts/train_svr_dcg.sh

Inference

  • test the DCGNet for Point Set AutoEncoding:
bash ./scripts/test_svr_dcg.sh
  • test the DCGNet for Point Set Reconstruction from a Single Image:
bash ./scripts/test_svr_dcg.sh

Citing this work

If you find this code useful for your research, please consider citing the following paper:

@inproceedings{ijcai2019-517,
  title     = {Deep Cascade Generation on Point Sets},
  author    = {Wang, Kaiqi and Chen, Ke and Jia, Kui},
  booktitle = {Proceedings of the Twenty-Eighth International Joint Conference on
               Artificial Intelligence, {IJCAI-19}},
  publisher = {International Joint Conferences on Artificial Intelligence Organization},
  pages     = {3726--3732},
  year      = {2019},
  month     = {7},
  doi       = {10.24963/ijcai.2019/517},
  url       = {https://doi.org/10.24963/ijcai.2019/517},
}

Acknowledgements

This work is supported in part by the Program for Guangdong Introducing Innovative and Enterpreneurial Teams (Grant No.: 2017ZT07X183), the National Natural Science Foundation of China (Grant No.: 61771201), and the Program of the Construction of Talented Personnel by the South China University of Technology (Grant No.: D6192110).

About

The code for IJCAI 2019 paper "Deep Cascade Generation on Point Sets"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published