LiDAR-based multi-task road perception network for autonomous vehicles

• Main Authors: Yan, Fuwu, Wang, Kewei, Zou, Bin, Tang, Luqi, Li, Wenbo, Lv, Chen
• Other Authors: School of Mechanical and Aerospace Engineering
• Format: Article
• Language: English
• Published: 2021
• Subjects: Engineering::Mechanical engineering; Autonomous Vehicles; Dense Height Estimation
• Online Access: https://hdl.handle.net/10356/145714
• Institution: Nanyang Technological University

For autonomous vehicles, it is an important requirement to obtain integrate static road information in real-time in dynamic driving environment. A comprehensive perception of the surrounding road should cover the accurate detection of the entire road area despite occlusion, the 3D geometry and the types of road topology in order to facilitate the practical applications in autonomous driving. To this end, we propose a lightweight and efficient LiDAR-based multi-task road perception network (LMRoadNet) to conduct occlusion-free road segmentation, road ground height estimation, and road topology recognition simultaneously. To optimize the proposed network, a corresponding multi-task dataset, named MultiRoad, is built semi-automatically based on the public SemanticKITTI dataset. Specifically, our network architecture uses road segmentation as the main task, and the remaining two tasks are directly decoded on a concentrated 1/4 scale feature map derived from the main task's feature maps of different scales and phases, which significantly reduces the complexity of the overall network while achieves high performance. In addition, a loss function with learnable weight of each task is adopted to train the neural network, which effectively balances the loss of each task and improves performance of the individual tasks. Extensive experiments on the test set show that the proposed network achieves great performance of the three tasks in real-time, outperforms the conventional multi-task architecture and is comparable to the state-of-the-art efficient methods. Finally, a fusion strategy is proposed to combine results on different directions to expand the field of view for practical applications.