/ASIS

Associatively Segmenting Instances and Semantics in Point Clouds

Primary LanguagePythonMIT LicenseMIT

Associatively Segmenting Instances and Semantics in Point Clouds

[arXiv]

Overview

Dependencies

The code has been tested with Python 2.7 on Ubuntu 14.04.

Data and Model

  • Download 3D indoor parsing dataset (S3DIS Dataset). Version 1.2 of the dataset is used in this work.
python collect_indoor3d_data.py
python gen_h5.py
cd data && python generate_input_list.py
cd ..
  • (optional) Prepared HDF5 data for training is available here.

  • (optional) Trained model can be downloaded from here.

Usage

  • Compile TF Operators

    Refer to PointNet++

  • Training

cd models/ASIS/
ln -s ../../data .
sh +x train.sh 5
  • Evaluation
python eval_iou_accuracy.py

Note: We test on Area5 and train on the rest folds in default. 6 fold CV can be conducted in a similar way.

Citation

If our work is useful for your research, please consider citing:

@inproceedings{wang2019asis,
	title={Associatively Segmenting Instances and Semantics in Point Clouds},
	author={Wang, Xinlong and Liu, Shu and Shen, Xiaoyong and Shen, Chunhua, and Jia, Jiaya},
	booktitle={CVPR},
	year={2019}
}

Acknowledgemets

This code largely benefits from following repositories: PointNet++, SGPN, DGCNN and DiscLoss-tf