csjinxie/PointNL

Cascaded Non-local Neural Network for Point Cloud Semantic Segmentation

Python

Cascaded Non-local Neural Network for Point Cloud Semantic Segmentation

by Mingmei Cheng, Le Hui, Jin Xie, Jian Yang and Hui Kong, details are in paper

Requirements:

environment:
```
Ubuntu 18.04
```
install python package:
```
./Anaconda3-5.1.0-Linux-x86_64.sh
```
install PyTorch :
```
conda install pytorch==1.2.0
```

Usage:

build the ops

cd lib/pointops && python setup.py install && cd ../../

Train

Download the S3IDS dataset and generate superpoints.

We will update the detailed scripts of data processing later. The generation of superpoints can also refer to SPGraph.
```
sh tool/train.sh DATASET_PATH
```
Test
```
sh tool/test.sh DATASET_PATH
```

Citation:

  @article{cheng2020cascaded,
    title={Cascaded Non-local Neural Network for Point Cloud Semantic Segmentation},
    author={Cheng, Mingmei and Hui, Le and Xie, Jin and Yang, Jian and Kong, Hui},
    booktitle={IROS},
    year={2020}
}