Awesome
GIPSO: Geometrically Informed Propagation for Online Adaptation in 3D LiDAR Segmentation [ECCV2022]
The official implementation of our work "GIPSO: Geometrically Informed Propagation for Online Adaptation in 3D LiDAR Segmentation".
Introduction
3D point cloud semantic segmentation is fundamental for autonomous driving but most approaches neglect how to deal with domain shift when handling dynamic scenes. This paper advances the state of the art in this research field. Our first contribution consists in analysing the new unexplored scenario of Source-Free Online Unsupervised Domain Adaptation (SF-OUDA). We experimentally show that state-of-the-art methods have a rather limited ability to adapt pre-trained deep network models to unseen domains in an online manner. Our second contribution is an approach that relies on adaptive self-training and geometric-feature propagation to adapt a pre-trained source model online without requiring either source data or target labels. Our third contribution is to study SF-OUDA in a challenging setup where source data is synthetic and target data is point clouds captured in the real world. We use the recent SynLiDAR dataset as a synthetic source and introduce two new synthetic (source) datasets, which can stimulate future synthetic-to-real autonomous driving research. Our experiments show the effectiveness of our segmentation approach on thousands of real-world point clouds.
:fire: For more information follow the PAPER link! :fire:
Authors: Cristiano Saltori, Evgeny Krivosheev, Stéphane Lathuilière, Nicu Sebe, Fabio Galasso, Giuseppe Fiameni, Elisa Ricci, Fabio Poiesi
News :bell:
- 7/2022: Synth4D has been released!
- 7/2022: GIPSO code is OUT!:fire:
- 7/2022: GIPSO is accepted to ECCV 2022!:fire: Our work is the first allowing source-free online and unsupervised adaptation for 3D LiDAR semantic segmentation!
Installation
The code has been tested with Docker (see Docker container below) with Python 3.8, CUDA 10.2/11.1, pytorch 1.8.0 and pytorch-lighting 1.4.1. Any other version may requireq to update the code for compatibility.
Pip/Venv/Conda
In your virtual environment follow MinkowskiEnginge. This will install all the base packages.
Additionally, you need to install:
- open3d 0.13.0
- KNN-CUDA
- pytorch-lighting 1.4.1
- wandb
- tqdm
- pickle
If you want to work on nuScenes you need to install
Docker container
If you want to use Docker you can find a ready-to-use container at crissalto/online-adaptation-mink:1.2, just be sure to have installed drivers compatible with CUDA 11.1.
Synth4D Dataset
To enable full compatibility with SemanticKITTI and nuScenes, we present Synth4D, which we created using the CARLA simulator.
Tab.1 compares Synth4D to the other synthetic datasets.
Synth4D is composed of two sets of point cloud sequences, one compatible with SemanticKITTI (Velodyne HDL64E)and one compatible with nuScenes (Velodyne HDL32E).
Each set is composed of 20K labelled point clouds.
Synth4D is captured using a vehicle navigating in four scenarios (town, highway, rural area and city).
Because UDA requires consistent labels between source and target, we mapped the labels of Synth4D with those of SemanticKITTI/nuScenes using the original instructions given to annotators, thus producing eight macro classes: vehicle, pedestrian, road, sidewalk, terrain, manmade, vegetation and unlabelled.
The dataset can be downloaded at the following links:
- Synthetic Velodyne HDL64E (Synth4D-KITTI)
- Synthetic Velodyne HDL32E (Synth4D-nuScenes)
- Training and validation splits (used in the paper)
Data preparation
Synth4D
Download the Synth4D dataset following the above instructions and prepare the dataset paths as follows:
./
├──
├── ...
└── path_to_data_shown_in_config/
├──kitti_synth/
| ├──Town03/
| | ├── calib/
| | | ├── 000000.npy
| | | └── ...
| | ├── labels/
| | | ├── 000000.npy
| | | └── ...
| | └── velodyne/
| | ├── 000000.npy
| | └── ...
| ├──Town06/
| ├──Town07/
| └──Town10HD/
├──nuscenes_synth/
└──splits/
SynLiDAR
Download SynLiDAR dataset from here, then prepare data folders as follows:
./
├──
├── ...
└── path_to_data_shown_in_config/
└──sequences/
├── 00/
│ ├── velodyne/
| | ├── 000000.bin
| | ├── 000001.bin
| | └── ...
│ └── labels/
| ├── 000000.label
| ├── 000001.label
| └── ...
└── 12/
SemanticKITTI
To download SemanticKITTI follow the instructions here. Then, prepare the paths as follows:
./
├──
├── ...
└── path_to_data_shown_in_config/
└── sequences
├── 00/
│ ├── velodyne/
| | ├── 000000.bin
| | ├── 000001.bin
| | └── ...
│ ├── labels/
| | ├── 000000.label
| | ├── 000001.label
| | └── ...
| ├── calib.txt
| ├── poses.txt
| └── times.txt
└── 08/
nuScenes
Follow the instructions here to download the data and paths will be already like that:
./
├──
├── ...
└── path_to_data_shown_in_config/
├──v1.0-trainval
├──v1.0-test
├──samples
├──sweeps
├──maps
└──lidarseg
If you need to save space on your server you can remove sweeps as they are not used.
After you downloaded the datasets you need, create soft-links in the data directory
cd gipso-sfouda
mkdir data
ln -s PATH/TO/SYNTH4D Synth4D
# do the same for the other datasets
Source training
To train the source model on Synth4D
python train_lighting.py --config_file configs/source/synth4dkitti_source.yaml
For SynLiDAR use --config_file configs/source/synlidar_source.yaml.
For nuScenes --config_file configs/source/synth4dnusc_source.yaml
NB: we provide pretrained models, so you can skip this time-consuming step!:rocket:
NB: the code uses wandb for logs. Follow the instructions here and update your config.wandb.project_name and config.wandb.entity_name.
Pretrained models
We provide pretrained models on Synth4D-KITTI, Synth4D-nuScenes and SynLIDAR. You can find the models here. For the model performance please refer to the main paper.
After downloading the pretrained models decompress them in gipso-sfouda/pretrained_models.
Preprocess geometric features
First we need to pre-compute geometric features by using DIP.
This step will use the pretrained model in pretrained_models/dip_model.
To compute geometric features on SemanticKITTI
python compute_dip_features_kitti.py --source_path PATH/TO/SEMANTICKITTI/IN/CONFIGS --split 0
python compute_dip_features_kitti.py --source_path PATH/TO/SEMANTICKITTI/IN/CONFIGS --split 1
python compute_dip_features_kitti.py --source_path PATH/TO/SEMANTICKITTI/IN/CONFIGS --split 2
python compute_dip_features_kitti.py --source_path PATH/TO/SEMANTICKITTI/IN/CONFIGS --split 3
You can also run in parallel on each split to save time.
To compute geometric features on nuScenes
python compute_dip_features_nuscenes.py --source_path PATH/TO/NUSCENES/IN/CONFIGS
This will save geometric features in experiments/dip_features/semantickitti and experiments/dip_features/nuscenes, respectively.
If you want to change features path add ---save_path PATH/TO/SAVE/FEATURES.
Adaptation to target
If you use W&B, you will need to update your config.wandb.project_name and config.wandb.entity_name.
To adapt the source model Synth4DKITTI to the target domain SemanticKITTI
CUBLAS_WORKSPACE_CONFIG=:4096:8 python adapt_online.py --config_file configs/adaptation/synth4d2kitti_adaptation.yaml --geometric_path experiments/dip_features/semantickitti
The adapted model will be saved following config file in pipeline.save_dir together with evaluation results.
If you want to save point cloud for future visualization you will need to add --save_predictions and they will be saved in pipeline.save_dir.
References
If you use our work, please cite us:
@inproceedings{saltori2022gipso,
title={GIPSO: Geometrically Informed Propagation for Online Adaptation in 3D LiDAR Segmentation},
author={Saltori, Cristiano and Krivosheev, Evgeny and Lathuili{\'e}re, St{\'e}phane and Sebe, Nicu and Galasso, Fabio and Fiameni, Giuseppe and Ricci, Elisa and Poiesi, Fabio},
booktitle={European Conference on Computer Vision},
pages={567--585},
year={2022},
organization={Springer}
}
Acknowledgments
The work was partially supported by OSRAM GmbH, by the Italian Ministry of Education, Universities and Research (MIUR) ”Dipartimenti di Eccellenza 2018-2022”, by the SHIELD project, funded by the European Union’s Joint Programming Initiative – Cultural Heritage, Conservation, Protection and Use joint call and, it was carried out in the Vision and Learning joint laboratory of FBK and UNITN.
Thanks
We thanks the open source projects DIP, Minkowski-Engine, and KNN-KUDA.