Awesome

DAPS3D: Domain Adaptive Projective Segmentation of 3D LiDAR Point Clouds

Alexey Klokov, Di Un Pak, Aleksandr Khorin, Dmitry Yudin, Leon Kochiev, Vladimir Luchinskiy, Vitaly Bezuglyj

[pdf] [BibTeX]

This repo contains the code for our paper DAPS3D: Domain Adaptive Projective Segmentation of 3D LiDAR Point Clouds.

Abstract

LiDARs are one of the key sources of reliable environmental ranging information for autonomous vehicles. However, segmentation of 3D scene elements (roads, buildings, people, cars, etc.) based on LiDAR point clouds has limitations. On the one hand, point- and voxel-based segmentation neural networks do not offer sufficiently high speed. On the other hand, modern labeled datasets primarily consist of street scenes recorded for driverless cars and contain little data for mobile delivery robots or cleaners that must work in parks and yards with heavy pedestrian traffic. This article aims to overcome these limitations. We have proposed a novel approach called DAPS3D to train deep neural networks for 3D semantic segmentation. This approach is based on a spherical projection of a point cloud and LiDAR-specific masks, enabling the model to adapt to different types of LiDAR. First of all, we have introduced various high-speed multi-scale spherical projection segmentation models, including convolutional, recurrent, and transformer architectures. Among them, the SalsaNextRecLSTM model has shown the best results. Secondly, we have proposed several original augmentations for spherical projections of LiDAR data, including FoV, flip, and rotation augmentation, as well as a special T-Zone cutout. These augmentations increase the model's invariance when dealing with changes in the data domain. Finally, we introduce a new method to generate synthetic datasets for domain adaptation problems. We have developed two new datasets for validating 3D scene outdoor segmentation algorithms: the DAPS-1 dataset, which is based on the augmentation of the reconstructed 3D semantic map, and the DAPS-2 LiDAR dataset, collected by the on-board sensors of a cleaning robot in a park area. Particular attention is given to the performance of the developed models, demonstrating their ability to function in real-time.

ToDo List

• Release code
• Document for Installation
• Document for Dataset Preparation
• Document for Testing and Training
• Evaluation
• Release trained models
• Release datasets: DAPS-1 & DAPS-2
• Release SalsaNextRecLSTM model
• Release SalsaNextRecLSTM model

Installation Instructions

• We use conda environments
• We use different Python and PyTorch versions for our models
• For complete installation instructions, please see Installation.

Dataset Preparation

• We release our DAPS-1 and DAPS-2 datasets.
• We also experiment on SemanticKITTI and RELLIS-3D datasets.
• Please see Datasets Preparation for complete instructions for preparing the datasets.

Execution Instructions

Training

• We train all our models using 2 × RTX 2080Ti (11 GB each) GPUs.
• Please see Getting Started for training commands.

Evaluation

• Please see Getting Started for evaluation commands.

Results

You can find our pretrained models in Getting Started.

DAPS-1

All models are trained with augmentation set 5 with T-Zone (see our Paper)

<br/><br/>

DAPS-2

These are inference results on DAPS-2 for the models trained on different datasets.

Models trained on SemanticKITTI

Models trained on SemanticKITTI + RELLIS-3D

Models trained on DAPS-1

Citation

If you found DAPS3D useful in your research, please consider starring ⭐ us on GitHub and citing 📚 us in your research!

@article{klokov2023daps3d,
  title={DAPS3D: Domain Adaptive Projective Segmentation of 3D LiDAR Point Clouds},
  author={Klokov, Alexey and Pak, Di Un and Khorin, Aleksandr and Yudin, Dmitry and Kochiev, Leon and Luchinskiy, Vladimir and Bezuglyj, Vitaly},
  journal={IEEE Access},
  year={2023},
  doi={10.1109/ACCESS.2023.3298706},
  publisher={IEEE}
}

Licence

This repository is released under MIT License (see LICENSE file for details).