LiDAR-MOS: Moving Object Segmentation in 3D LiDAR Data
This repo contains the code for our paper: Moving Object Segmentation in 3D LiDAR Data: A Learning-based Approach Exploiting Sequential Data PDF.
Our approach accurately segments the scene into moving and static objects, i.e., distinguishing between moving cars vs. parked cars. It runs faster than the frame rate of the sensor and can be used to improve 3D LiDAR-based odometry/SLAM and mapping results as shown below.
Additionally, we created a new benchmark for LiDAR-based moving object segmentation based on SemanticKITTI here.
Complete demo video can be found in YouTube here. LiDAR-MOS in action:
Table of Contents
- Introduction of the repo and benchmark
- Publication
- Dependencies
- How to use
- Applications
- Collection of downloads
- License
Publication
If you use our code and benchmark in your academic work, please cite the corresponding paper:
@article{chen2021ral,
title={{Moving Object Segmentation in 3D LiDAR Data: A Learning-based Approach Exploiting Sequential Data}},
author={X. Chen and S. Li and B. Mersch and L. Wiesmann and J. Gall and J. Behley and C. Stachniss},
year={2021},
journal={IEEE Robotics and Automation Letters (RA-L)},
doi = {10.1109/LRA.2021.3093567},
issn = {2377-3766},
}
Dependencies
We built and tested our work based on SalsaNext, RangeNet++ and MINet. We thank the original authors for their nice work and implementation. If you are interested in fast LiDAR-based semantic segmentation, we strongly recommend having a look at the original repositories.
Note that, in this repo, we show that how easily we could achieve LiDAR-based moving object segmentation exploiting sequential information with existing segmentation networks. We didn't change the original pipeline of the segmentation networks, but only changed the data loader and input of the network as shown in the figure below. Therefore, our method can be used with any range-image-based LiDAR segmentation networks.
Our method is based on range images. To use range projection with fast c++ library, please find the usage doc here.
How to use
For a quick test of all the steps below, one could download a toy dataset here and decompress it in the data\ folder following the data structure data/README.md.
Prepare training data
To use our method, one needs to generate the residual images. Here is a quick demo:
$ python3 utils/gen_residual_images.py
More setup about the data preparation can be found in the yaml file config/data_preparing.yaml. To prepare the training data for the whole KITTI-Odometry dataset, please download the original website.
Using SalsaNext as the baseline
To use SalsaNext as the baseline segmentation network for LiDAR-MOS, one should follow the mos_SalsaNext/README.md to it up.
Inferring
To generate the LiDAR-MOS predictions with pretrained model. Quick test on toy dataset, directly run
$ cd mos_SalsaNext/train/tasks/semantic
$ python3 infer.py -d ../../../../data -m ../../../../data/model_salsanext_residual_1 -l ../../../../data/predictions_salsanext_residual_1_new -s valid
Inferring the whole dataset, please download the KITTI-Odometry dataset from the original website, and change the corresponding paths.
$ cd mos_SalsaNext/train/tasks/semantic
$ python3 infer.py -d path/to/kitti/dataset -m path/to/pretrained_model -l path/to/log -s train/valid/test # depending of desired split to evaluate
Training
To train a LiDAR-MOS network with SalsaNext from scratch, one has to download the KITTI-Odometry dataset and Semantic-Kitti dataset: Change the corresponding paths and run:
$ cd mos_SalsaNext/train/tasks/semantic
$ ./train.sh -d path/to/kitti/dataset -a salsanext_mos.yml -l path/to/log -c 0 # the number of used gpu cores
Using RangeNet++ as the baseline
To use RangeNet++ as the baseline segmentation network for LiDAR-MOS, one should follow the mos_RangeNet/README.md to set it up.
Inferring
Inferring the whole dataset, please download the KITTI-Odometry dataset from the original website, the [pretrained model](todo: add pretrained model for rangenet) and change the corresponding paths.
$ cd mos_RangeNet/tasks/semantic
$ python3 infer.py -d path/to/kitti/dataset -m path/to/pretrained_model -l path/to/log -s train/valid/test # depending of desired split to evaluate
Training
To train a LiDAR-MOS network with RangeNet++ from scratch, one has to download the KITTI-Odometry dataset and Semantic-Kitti dataset and change the corresponding paths and run:
$ cd mos_RangeNet/tasks/semantic
$ python3 train.py -d path/to/kitti/dataset -ac rangenet_mos.yaml -l path/to/log
More pretrained model and LiDAR-MOS predictions can be found in collection of downloads.
Evaluation and visualization
How to evaluate
Evaluation metrics. Let's call the moving (dynamic) status as D and the static status as S.
Since we ignore the unlabelled and invalid status, therefore in MOD there are only two classes.
| GT\Prediction | dynamic | static |
|---|---|---|
| dynamic | TD | FS |
| static | FD | TS |
- $$ IoU_{MOS} = \frac{TD}{TD+FD+FS} $$
To evaluate the MOS results on the toy dataset just run:
$ python3 utils/evaluate_mos.py -d data -p data/predictions_salsanext_residual_1_valid -s valid
To evaluate the MOS results on our LiDAR-MOS benchmark please have a look at our semantic-kitti-api and benchmark website.
How to visualize the predictions
To visualize the MOS results on the toy dataset just run:
$ python3 utils/visualize_mos.py -d data -p data/predictions_salsanext_residual_1_valid -s 8 # here we use a specific sequence number
where:
sequenceis the sequence to be accessed.datasetis the path to the kitti dataset where the sequences directory is.
Navigation:
nis next scan,bis previous scan,escorqexits.
Applications
LiDAR-MOS is very important for building consistent maps, making future state predictions, avoiding collisions, and planning. It can also improve and robustify pose estimation, sensor data registration, and SLAM. Here we show two obvious applications of our LiDAR-MOS which are LiDAR-based odometry/SLAM as well as 3D mapping. Before that, we show two simple examples of how to combine our method with semantics and clean the scans. After cleaning scans we can get better odometry/SLAM and 3D mapping results.
Note that, here we show two direct use cases of our MOS approach without any further optimizations employed.
Enhanced with semantics
To show a simple way of combining our LiDAR-MOS with semantics, we provide a quick demo with the toy dataset:
$ python3 utils/combine_semantics.py
It just simply checks whether the moving objects are movable classes or not. If not, re-assigned as static.
Clean the scans
To clean the LiDAR scans with our LiDAR-MOS as masks, we also provide a quick demo on the toy dataset:
$ python3 utils/scan_cleaner.py
Odometry/SLAM
Using the cleaned LiDAR scans, we see that by simply applying our MOS predictions as a preprocessing mask, the odometry results are improved in both the KITTI training and test data and even slightly better than the carefully-designed full classes semantic-enhanced SuMa++.
The testing results of our methods can also be found in KITTI-Odometry benchmark.
Mapping
we compare the aggregated point cloud maps (left) directly with the raw LiDAR scans, (right) with the cleaned LiDAR scans by applying our MOS predictions as masks. As can be seen, there are moving objects present that pollute the map, which might have adversarial effects, when used for localization or path planning. By using our MOS predictions as masks, we can effectively remove these artifacts and get a clean map.
Collection of downloads
- LiDAR_MOS_toy_dataset (toy dataset used for the quick demos)
- predictions_salsanext_semantic (semantic segmentation results from SalsaNext for all sequences 00 - 21)
- predictions_salsanext_residual_8_sem (Our best! LiDAR-MOS results using SalsaNext with 8 residual images + semantics)
- model_rangenet_residual_1 (pretrained model using RangeNet++ with 1 residual image)
- model_salsanext_residual_1 (pretrained model using SalsaNext with 1 residual image)
- model_salsanext_residual_8 (pretrained model using SalsaNext with 8 residual image)
License
This project is free software made available under the MIT License. For details see the LICENSE file.
After running this command, the program seems to be stuck. I don't know why, I want to get your visualized results, as shown below:
ps：Author reply：From the results of the operation, it seems to be a tkinter problem.But the tkinker module does not seem to be missing.
If anyone knows how to solve it, hope you can help me, thanks.
35 Dec 14, 2022
1.4k Jan 07, 2023
62 Dec 21, 2022
4 Dec 31, 2022
77 Dec 07, 2022
16.5k Jan 08, 2023
10 Dec 09, 2022
488 Nov 28, 2022
805 Jan 09, 2023
6 Sep 09, 2021
104 Nov 12, 2022
28 Jan 01, 2023
551 Jan 07, 2023
11 Nov 15, 2022
5 Nov 14, 2022
28 Dec 23, 2022
34 Sep 06, 2022
348 Jan 05, 2023
2.3k Jan 01, 2023
2 Mar 30, 2022