AgentI2P: Optimizing Image-to-Point Cloud Registration Via Behaviour Cloning and Reinforcement Learning.

Shen Yan,Maojun Zhang,Yang Peng,Yu Liu,Hanlin Tan
DOI: https://doi.org/10.3390/rs14246301
IF: 5
2022-01-01
Remote Sensing
Abstract:Image-to-point cloud registration refers to finding relative transformation between the camera and the reference frame of the 3D point cloud, which is critical for autonomous driving. Recently, a two-stage “frustum point cloud classification + camera pose optimization” pipeline has shown impressive results on this task. This paper focuses on the second stage and reformulates the optimization procedure as a Markov decision process. An initial pose is modified incrementally, sequentially aligning a virtual 3D point observation towards a previous classification solution. We consider such an iterative update process as a reinforcement learning task and, to this end, propose a novel agent (AgentI2P) to conduct decision making. To guide AgentI2P, we employ behaviour cloning (BC) and reinforcement learning (RL) techniques: cloning an expert to learn accurate pose movement and reinforcing an alignment reward to improve the policy further. [We demonstrate the effectiveness and efficiency of our approach on Oxford Robotcar and KITTI datasets. The (RTE, RRE) metrics are (1.34m,1.46∘) on Oxford Robotcar and (3.90m,5.94∘) on KITTI, and the inference time is 60 ms, both achieving state-of-the-art performance]. The source code will be publicly available upon publication of the paper.
What problem does this paper attempt to address?