DQN-based gradual fisheye image rectification

Jie Zhao,Shikui Wei,Lixin Liao,Yao Zhao
DOI: https://doi.org/10.1016/j.patrec.2021.08.025
IF: 4.757
2021-12-01
Pattern Recognition Letters
Abstract:Fisheye image rectification is a classic and important task in the computer vision area, which is generally treated as a pre-processing step in many application scenarios. Most of the existing fisheye image rectification methods focused mainly on building a direct (one-step) projection relationship between fisheye images and corrected images. Although these methods have achieved impressive performance, they depended heavily on data distribution and cannot work well on images whose distortion parameters are out of range. To address this issue, we propose a multi-step gradual image rectification scheme. In particular, we treat the fisheye image rectification problem as one Markov Decision Process and employ a widely-used deep reinforcement learning method (i.e., Deep Q-Network) to solve the problem. In addition, we build a new large-scale fisheye image dataset to evaluate our method, which contains 50,000 training images and 5,000 test images. Experimental results show the proposed method can better handle images with a wide range of distortion.
computer science, artificial intelligence
What problem does this paper attempt to address?