Need help with Grid-GCN?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

Xharlie
159 Stars 20 Forks 35 Commits 8 Opened issues

Description

Grid-GCN for Fast and Scalable Point Cloud Learning

Services available

!
?

Need anything else?

Contributors list

# 237,004
Python
C++
Shell
pointne...
32 commits

Grid-GCN for Fast and Scalable Point Cloud Learning (CVPR2020)

Please cite us:

@article{1912.02984,
  Author = {Qiangeng Xu and Xudong Sun and Cho-Ying Wu and Panqu Wang and Ulrich Neumann},
  Title = {Grid-GCN for Fast and Scalable Point Cloud Learning},
  Year = {2019},
  Eprint = {arXiv:1912.02984},
  Howpublished = {Proceedings of the IEEE Conference on Computer Vision and Pattern
    Recognition (CVPR 2020)}
}

Requirement: GGCN implemented by MXNET 1.5.0

make sure you have gcc version suggested by MXNET 1.5.0

Install Our CUDA modules to MXNET Libary:

cd gridifyop
vim Makefile  # then change mx_home to your mxnet-apache directory, and adjust nvcc command according to your gpu model and cuda version. here we use compute power 61 and 75 for 1080 ti and 2080 ti. save the change
make
cd ..

Data Preparation

  • Classification

    • #### ModelNet40 We refer to pointnet https://github.com/charlesq34/pointnet/blob/master/provider.py
      cd data/
      wget https://shapenet.cs.stanford.edu/media/modelnet40_ply_hdf5_2048.zip
      unzip modelnet40_ply_hdf5_2048.zip
      unzip it and put it inside data/
      
    • #### ModelNet10 please refer to pointnet++'s github
      download  modelnet40_normal_resampled from https://github.com/charlesq34/pointnet2
      take the modelnet10_train.txt, modelnet10_test.txt and extract from modelnet40_ply_hdf5_2048 to create a modelnet10_ply_hdf5_2048
      or use modelnet40_normal_resampled directly, but configuration file configs_10.yaml new: True -> False
      
  • Segmentation/ScanNet

    Please refer to pointnet++ for downloading ScanNet use link: ```

    in data/

    wget https://shapenet.cs.stanford.edu/media/scannetdatapointnet2.zip unzip scannetdatapointnet2.zip mv data scannet

Training

  • Classification

    • #### ModelNet40 ``` cd classification nohup python -u train/traingpuggcn_mdl40.py &> mdl40.log &
  * #### ModelNet10
  please refer to pointnet++

cd classification nohup python -u train/traingpuggcn_mdl10.py &> mdl10.log &

* ### Segmentation 
  * #### ScanNet
  Please refer to pointnet++ for downloading ScanNet use link: 

cd segmentation

### then you cd configs -> go to configs.yaml to choose 8192 points model or 81920 points model by leaving one of them uncommented

nohup python -u traintest/trainggcn_scannet.py &> train.log & ```

Testing

  • ### Segmentation
    • #### ScanNet ``` cd segmentation

### then you cd configs -> go to configs.yaml to choose 8192 points model or 81920 points model by leaving one of them uncommented ### you should also change loadmodelprefix to the intented trained model file in your output directory.

nohup python -u traintest/testggcn_scannet.py &> test.log & ```

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.