Home

Awesome

arXiv GitHub Stars visitors

Towards Semantic Segmentation of Urban-Scale 3D Point Clouds: A Dataset, Benchmarks and Challenges

This is the official repository of the SensatUrban dataset. For technical details, please refer to:

Towards Semantic Segmentation of Urban-Scale 3D Point Clouds: A Dataset, Benchmarks and Challenges <br /> Qingyong Hu, Bo Yang*, Sheikh Khalid, Wen Xiao, Niki Trigoni, Andrew Markham. <br /> [Paper] [Blog] [Video] [Project page] [Download] [Evaluation] [Urban3D workshop@ECCV2022] <br />

We are hosting the 2nd Urban3D Challenge at ECCV 2022, please refer to this page for more details.

(1) Dataset

1.1 Overview

This dataset is an urban-scale photogrammetric point cloud dataset with nearly three billion richly annotated points, which is five times the number of labeled points than the existing largest point cloud dataset. Our dataset consists of large areas from two UK cities, covering about 6 km^2 of the city landscape. In the dataset, each 3D point is labeled as one of 13 semantic classes, such as ground, vegetation, car, etc..

<p align="center"> <img src="imgs/Fig1.png" width="100%"> </p> <p align="center"> <img src="imgs/Table1.png" width="100%"> </p>

1.2 Data Collection

The 3D point clouds are generated from high-quality aerial images captured by a professional-grade UAV mapping system. In order to fully and evenly cover the survey area, all flight paths are pre-planned in a grid fashion and automated by the flight control system (e-Motion).

<p align="center"> <img src="imgs/Fig2.png" width="70%"> </p>

1.3 Semantic Annotations

<p align="center"> <img src="imgs/Fig3.png" width="100%"> </p>

1.4 Statistics

<p align="center"> <img src="imgs/Fig5.png" width="100%"> </p>

(2) Benchmarks

We extensively evaluate the performance of state-of-the-art algorithms on our dataset and provide a comprehensive analysis of the results. In particular, we identify several key challenges towards urban-scale point cloud understanding.

<p align="center"> <img src="imgs/Fig6.png" width="100%"> </p>

(3) Demo

<p align="center"> <a href="https://youtu.be/IG0tTdqB3L8"><img src="http://point-cloud-analysis.cs.ox.ac.uk/imgs/3DV_demo_cover.png" width="80%"></a> </p>

(4) Training and Evaluation

Here we provide the training and evaluation script of RandLA-Net for your reference.

Download the files named "data_release.zip" here. Uncompress the folder and move it to /Dataset/SensatUrban.

conda create -n randlanet python=3.5
source activate randlanet
pip install -r helper_requirements.txt
sh compile_op.sh
python input_preparation.py --dataset_path $YOURPATH
cd $YOURPATH; 
cd ../; mkdir original_block_ply; mv data_release/train/* original_block_ply; mv data_release/test/* original_block_ply;
mv data_release/grid* ./

The data should organized in the following format:

/Dataset/SensatUrban/
          └── original_block_ply/
                  ├── birmingham_block_0.ply
                  ├── birmingham_block_1.ply 
		  ...
	    	  └── cambridge_block_34.ply 
          └── grid_0.200/
	     	  ├── birmingham_block_0_KDTree.pkl
                  ├── birmingham_block_0.ply
		  ├── birmingham_block_0_proj.pkl 
		  ...
	    	  └── cambridge_block_34.ply 
python main_SensatUrban.py --mode train --gpu 0 
python main_SensatUrban.py --mode test --gpu 0 
<p align="center"> <img src="imgs/wechat_code.jpg" width="50%"> </p>

Citation

If you find our work useful in your research, please consider citing:

@inproceedings{hu2020towards,
  title={Towards Semantic Segmentation of Urban-Scale 3D Point Clouds: A Dataset, Benchmarks and Challenges},
  author={Hu, Qingyong and Yang, Bo and Khalid, Sheikh and Xiao, Wen and Trigoni, Niki and Markham, Andrew},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  year={2021}
}


@article{hu2022sensaturban,
title={Sensaturban: Learning semantics from urban-scale photogrammetric point clouds},
author={Hu, Qingyong and Yang, Bo and Khalid, Sheikh and Xiao, Wen and Trigoni, Niki and Markham, Andrew},
journal={International Journal of Computer Vision},
volume={130},
number={2},
pages={316--343},
year={2022},
publisher={Springer}

}

Updates

Related Repos

  1. RandLA-Net: Efficient Semantic Segmentation of Large-Scale Point Clouds GitHub stars
  2. SoTA-Point-Cloud: Deep Learning for 3D Point Clouds: A Survey GitHub stars
  3. 3D-BoNet: Learning Object Bounding Boxes for 3D Instance Segmentation on Point Clouds GitHub stars
  4. SpinNet: Learning a General Surface Descriptor for 3D Point Cloud Registration GitHub stars
  5. SQN: Weakly-Supervised Semantic Segmentation of Large-Scale 3D Point Clouds GitHub stars
  6. Not All Points Are Equal: Learning Highly Efficient Point-based Detectors for 3D LiDAR Point Clouds GitHub stars
  7. STPLS3D: A Large-Scale Synthetic and Real Aerial Photogrammetry 3D Point Cloud Dataset GitHub stars