Donwload prepared HDF5 data for training:
sh download_data.sh
(optional) Download 3D indoor parsing dataset (S3DIS Dataset) for testing and visualization. Version 1.2 of the dataset is used in this work.
To prepare your own HDF5 data, you need to firstly download 3D indoor parsing dataset and then use python collect_indoor3d_data.py
for data re-organization and python gen_indoor3d_h5.py
to generate HDF5 files.
ln -s ../data/indoorstandfordh5_data/
python train.py
python batch_inference.py
python eval_iou_accuracy.py
mIOU=sum(mIOU(model1+...+model6))/6
To evaluate overall segmentation accuracy, we evaluate 6 models on their corresponding test areas and use eval_iou_accuracy.py
to produce point classification accuracy and IoU as reported in the paper.