This repository is our project for the course Point Cloud for 3D Modeling (NPM3D) of the IASD master program. This implementation is strongly inspired from the paper classification of aerial point clouds with deep learning by Emre Ă–zdemir and Fabio Remondino.
Our project report is available at the following link : https://www.overleaf.com/read/tdjgnxjbcmpg
This project requires python >= 3.5
$ git clone https://github.com/theobdt/aerial_pc_classification.git
$ cd aerial_pc_classification
$ pip3 install -r requirements.txt
To get the data, you will first have to request access on the ISPRS website, it is totally free. Fill out the questionnaire to receive the credentials. Then download data using the following script :
$ chmod +x fetch_data.sh
$ ./fetch_data.sh
Convert files to PLY and center/scale the point clouds.
$ python3 preprocessing.py -f data/vaihingen3D_train.pts data/vaihingen3D_test.pts --centering
Output files will be saved to data/preprocessed/
.
Parameters used for features extraction are stored in cfg/config_features_extraction.yaml
.
Output files will be saved in data/features/
.
Available steps are :
- Local descriptors (
descriptors
). A few examples :Normal angles to vertical axis
Sphericity
Planarity
- Region growing (
region_growing
)Results of region growing
- Ground extraction (
ground_extraction
)Ground extraction
- Ground rasterization (
ground_rasterization
)Ground rasterized
- Height above ground (
height_above_ground
)Height above ground
To run all steps at once:
$ python3 compute_features.py --full_pipeline --files data/preprocessed/vaihingen3D_train.ply data/preprocessed/vaihingen3D_test.ply
To run all steps starting from a specific one :
$ python3 compute_features.py --from_step region_growing --files data/preprocessed/vaihingen3D_train.ply data/preprocessed/vaihingen3D_test.ply
To run steps individually :
$ python3 compute_features.py --steps region_growing ground_extraction --files data/preprocessed/vaihingen3D_train.ply data/preprocessed/vaihingen3D_test.ply
Configuration file for training is stored in cfg/config_bilstm.yaml
.
To train the network, run the following command :
$ python3 train.py
A new checkpoint will be created in ckpts/[current timestamp]
, along with a plot of accuracies and losses throughout training.
To compute predictions on all points of the cloud run :
$ python3 test.py -f data/features/vaihingen3D_test.ply --ckpt ckpts/2020-04-10_12-25-36
A new folder will be created at data/predictions/[model timestamp]
, along with a comprehensive metrics table.
9-classes test set
9-classes predictions on test set