Online Range Image-based Pole Extractor for Long-term LiDAR Localization in Urban Environments

The code for the learning-based pole extractor will come soon.

This repo contains the code for our ECMR2021 paper Online Range Image-based Pole Extractor for Long-term LiDAR Localization in Urban Environments and RAS submitted paper Online Pole Segmentation on Range Images for Long-term LiDAR Localization in Urban Environments.

Developed by Hao Dong and Xieyuanli Chen.

Overview of our approach. A. we project the LiDAR point cloud into a range image and B. extract poles in the image. C. based on the extracted poles, we then build a global pole map of the environment. D. we finally propose a pole-based observation model for MCL to localize the robot in the map.

Publication

If you use our implementation in your academic work, please cite the corresponding conference paper and journal paper:

@InProceedings{dong2021ecmr,
    author = {H. Dong and X. Chen and C. Stachniss},
    title = {{Online Range Image-based Pole Extractor for Long-term LiDAR Localization in Urban Environments}},
    booktitle = {Proceedings of the European Conference on Mobile Robots (ECMR)},
    year = {2021}
    }

@article{dong2022ras,
author   = {H. Dong and X. Chen and S. Särkkä and C. Stachniss},
title    = {{Online Pole Segmentation on Range Images for Long-term LiDAR Localization in Urban Environments}},
journal  = arxiv,
eprint   = {2208.07364},
year     = {2022},
}

Dependencies

The code was tested with Ubuntu 20.04 with its standard python version 3.8.

• Python dependencies

  pip install numpy matplotlib open3d-python progressbar2 pyquaternion transforms3d scipy scikit-image networkx numba arrow pykitti

How to use

NCLT Dataset

Download the dataset and extract the data in the /nclt/data folder following the recommended data structure, and then run:

python src/ncltpoles.py

KITTI Dataset

Download the KITTI raw data 2011_09_26_drive_0009 by navigating to the /kitti/raw_data folder and run:

./kitti_downloader.sh

then run:

python src/kittipoles.py

MulRan Dataset

Download the KAIST 01 and KAIST 02 dataset and extract the data in the /mulran/data folder following the recommended data structure, and then run:

python src/mulranpoles.py

Pole Dataset

The pole datasets are stored in the /data/pole-dataset/KITTI and /data/pole-dataset/NCLT folders. The data are stored in .npz format with the shape N*2. Each row represents the x and y position of one pole. You can evaluate the pole extraction with the groud-truth pole map by running:

python src/test_match.py

License

Copyright 2021, Hao Dong, Xieyuanli Chen, Cyrill Stachniss, Photogrammetry and Robotics Lab, University of Bonn.

This project is free software made available under the MIT License. For details see the LICENSE file.

Acknowledgement

Many thanks to the excellent open-source projects polex and SalsaNext.