OG-Net

I have modified code from DGCNN https://github.com/WangYueFt/dgcnn

cd pytorch

Install

I also modified code from PointNet++, so you are recommended to install the packages.

https://github.com/layumi/person-reid-3d#install

Train

python main.py --exp_name=ognet_1024_d2_efficient_small_e500  --model=ognet --dropout 0 --feature_dims 48,96,192,384  --efficient  --epoch 500 --dropout 0.2

Evaluate

python main.py --exp_name=ognet_1024_d2_efficient_small_e500 --model=ognet  --eval=True  --model_path=checkpoints/ognet_1024_d2_efficient_small_e500/models/model.t7

Result is 93.35 / 90.55

Dynamic Graph CNN for Learning on Point Clouds

We propose a new neural network module dubbed EdgeConv suitable for CNN-based high-level tasks on point clouds including classification and segmentation. EdgeConv is differentiable and can be plugged into existing architectures.

[Project] [Paper]

Overview

DGCNN is the author's re-implementation of Dynamic Graph CNN, which achieves state-of-the-art performance on point-cloud-related high-level tasks including category classification, semantic segmentation and part segmentation.

Further information please contact Yue Wang and Yongbin Sun.

Author's Implementations

The classification experiments in our paper are done with the pytorch implementation.

Other Implementations

pytorch-geometric
pytorch-dgcnn (This implementation on S3DIS achieves significant better results than our tensorflow implementation)

Citation

Please cite this paper if you want to use it in your work,

@article{dgcnn,
  title={Dynamic Graph CNN for Learning on Point Clouds},
  author={Wang, Yue and Sun, Yongbin and Liu, Ziwei and Sarma, Sanjay E. and Bronstein, Michael M. and Solomon, Justin M.},
  journal={ACM Transactions on Graphics (TOG)},
  year={2019}
}

License

MIT License

Acknowledgement

The structure of this codebase is borrowed from PointNet.

layumi / dgcnn