Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
liuziwei7 committed Feb 10, 2018
1 parent 9f1e9d6 commit 66ce7a4
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 7 deletions.
36 changes: 29 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,11 +1,30 @@
## Dynamic Graph CNN for Learning on Point Clouds
This is the implementation of DGCNN and heavily based on <a href="https://github.com/charlesq34/pointnet" target="_blank"> https://github.com/charlesq34/pointnet </a>
# 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.

### Classification
1. python train.py
2. After training finishes, python evalutate.py
[[Project]](https://liuziwei7.github.io/projects/DGCNN) [[Paper]](https://arxiv.org/abs/1801.07829)

### Citation
## 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.

<img src='./misc/demo_teaser.png' width=800>

Further information please contact [Yue Wang](https://www.csail.mit.edu/person/yue-wang) and [Yongbin Sun](https://autoid.mit.edu/people-2).

## Requirements
* [TensorFlow](https://www.tensorflow.org/)

## Point Cloud Classification
* Run the training script:
``` bash
python train.py
```
* Run the evaluation script after training finished:
``` bash
python evalutate.py

```

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

@article{dgcnn,
Expand All @@ -15,5 +34,8 @@ Please cite this paper if you want to use it in your work,
year={2018}
}

### License
## License
MIT License

## Acknowledgement
This code is heavily borrowed from [PointNet](https://github.com/charlesq34/pointnet).
Binary file added misc/demo_teaser.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 66ce7a4

Please sign in to comment.