sshan-zhao / FPConv

FPConv: Learning Local Flattening for Point Convolution, CVPR 2020

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FPConv

Yiqun Lin, Zizheng Yan, Haibin Huang, Dong Du, Ligang Liu, Shuguang Cui, Xiaoguang Han, "FPConv: Learning Local Flattening for Point Convolution", CVPR 2020 [paper]

@InProceedings{lin_fpconv_cvpr2020,
    author = {Yiqun Lin, Zizheng Yan, Haibin Huang, Dong Du, Ligang Liu, Shuguang Cui, Xiaoguang Han},
    title = {FPConv: Learning Local Flattening for Point Convolution},
    booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
    month = {June},
    year = {2020}
}

Introduction

We introduce FPConv, a novel surface-style convolution operator designed for 3D point cloud analysis. Unlike previous methods, FPConv doesn't require transforming to intermediate representation like 3D grid or graph and directly works on surface geometry of point cloud. To be more specific, for each point, FPConv performs a local flattening by automatically learning a weight map to softly project surrounding points onto a 2D grid. Regular 2D convolution can thus be applied for efficient feature learning. FPConv can be easily integrated into various network architectures for tasks like 3D object classification and 3D scene segmentation, and achieve comparable performance with existing volumetric-type convolutions. More importantly, our experiments also show that FPConv can be a complementary of volumetric convolutions and jointly training them can further boost overall performance into state-of-the-art results.

fig3

Installation

This code has been tested with Python 3.6, PyTorch 1.2.0, CUDA 10.0 and CUDNN 7.4 on Ubuntu 18.04.

Firstly, install pointnet2 by running the following commands:

cd fpconv/pointnet2
python setup.py install
cd ../

You may also need to install plyfile and pickle for data preprocessing.

Usage

Edit the global configuration file config.json before training.

{
    "version": "0.0",
    "scannet_raw": "<path_to_this_repo>/dataset/scannet_v2",
    "scannet_pickle": "<path_to_this_repo>/dataset/scannet_pickles",
    "scene_list": "<path_to_this_repo>/utils/scannet_datalist",
    "s3dis_aligned_raw": "<path_to_this_repo>/dataset/Stanford3dDataset_v1.2_Aligned_Version",
    "s3dis_data_root": "<path_to_this_repo>/dataset/s3dis_aligned"
}

— Semantic Segmentation on ScanNet

0. Baseline

Tested on eval split of ScanNet. (see ./utils/scannet_datalist/scannetv2_eval.txt)

Model mIoU mA oA download
fpcnn_scannet 64.4 76.4 85.8 ckpt-17.8M

1. Preprocessing

Download ScanNet v2 dataset to ./dataset/scannet_v2. Only _vh_clean_2.ply and _vh_clean_2.labels.ply should be downloaded for each scene. Specify the dataset path and output path in config.json and then run following commands for data pre-processing.

cd utils
python collect_scannet_pickle.py

It will generate 3 pickle files (scannet_<split>_rgb21c_pointid.pickle) for 3 splits (train, eval, test) respectively. We also provide a pre-processed ScanNet v2 dataset for downloading: Google Drive. The ./dataset folder should be organized as follows.

FPConv
├── dataset
│   ├── scannet_v2
│   │  ├── scans
│   │  ├── scans_test
│   ├── scannet_pickles
│   │  ├── scannet_train_rgb21c_pointid.pickle
│   │  ├── scannet_eval_rgb21c_pointid.pickle
│   │  ├── scannet_test_rgb21c_pointid.pickle

2. Training

Run the following command to start the training. Output (logs) will be redirected to ./logs/fp_scannet/nohup.log.

bash train_scannet.sh

We trained our model with 2 Titan Xp GPUs with batch size of 12. If you don't have enough GPUs for training, please reduce batch_size to 6 for single GPU.

3. Evaluation

Run the following command to evaluate model on evaluation dataset (you may need to modify the epoch in ./test_scannet.sh). Output (logs) will be redirected to ./test/fp_scannet_240.log.

bash test_scannet.sh

Note: Final evaluation (by running ./test_scannet.sh) is conducted on full point cloud, while evaluation during the training phase is conducted on randomly sampled points in each block of input scene.

— Semantic Segmentation on S3DIS

0. Baseline

Trained on Area 1~4 and 6, tested on Area 5.

Model mIoU mA oA download
fpcnn_s3dis 62.7 70.3 87.5 ckpt-70.0M

1. Preprocessing

Firstly, you need to download S3DIS dataset from link, aligned version 1.2 of the dataset is used in this work. Unzip S3DIS dataset into ./dataset/Stanford3dDataset_v1.2_Aligned_Version, and specify the dataset path and output path in config.json. Then run the following commands for pre-processing.

cd utils
python collect_indoor3d_data.py
cd ..

It will generate a .npy files for each room. The dataset folder should be organized as follows. We also provide a pre-processed S3DIS dataset for downloading: Google Drive.

FPConv
├── dataset
│   ├── Stanford3dDataset_v1.2_Aligned_Version
│   │  ├── Area_1
│   │  ├── ...
│   │  ├── Area_6
│   ├── s3dis_aligned
│   │  ├── *.npy

2. Training

Run the following command to start the training. Output (logs) will be redirected to ./logs/fp_s3dis/nohup.log.

bash train_s3dis.sh

We trained our model on S3DIS with 4 Titan Xp GPUs, batch size of 8 totally, and 100 epochs.

3. Evaluation

Run the following command to evaluate model on evaluation dataset (you may need to modify the epoch in ./test_s3dis.sh). Output (logs) will be redirected to ./test/fp_s3dis_60.log.

bash test_s3dis.sh

Acknowledgement

License

This repository is released under MIT License (see LICENSE file for details).

About

FPConv: Learning Local Flattening for Point Convolution, CVPR 2020

License:MIT License


Languages

Language:Python 84.8%Language:Cuda 9.3%Language:C++ 5.3%Language:Shell 0.6%