AUTOMATIC EXTRACTION OF ROAD CENTERLINES AND EDGE LINES FROM AERIAL IMAGES VIA CNN-BASED REGRESSION
- School of Remote Sensing and Information Engineering, Wuhan University, China
Keywords: Road extraction, Convolution Neural Network, Regression, Confidence map, Road Width
Abstract. Extracting roads from aerial images is a challenging task in the field of remote sensing. Most approaches formulate road extraction as a segmentation problem and use thinning and edge detection to obtain road centerlines and edge lines, which could produce spurs around the extracted centerlines/edge lines. In this study, a novel regression-based method is proposed to extract road centerlines and edge lines directly from aerial images. The method consists of three major steps. First, an end-to-end regression network based on CNN is trained to predict confidence maps for road centerlines and estimate road width. Then, after the CNN predicts the confidence map, non-maximum suppression and road tracking are applied to extract accurate road centerlines and construct road topology. Meanwhile, Road edge lines are generated based on the road width estimated by the CNN. Finally, in order to improve the connectivity of extracted road network, tensor voting is applied to detect road intersections and the detected intersections are used as guidance for the overcome of discontinuities. The experiments conducted on the SpaceNet and DeepGlobe datasets show that our approach achieves better performance than other methods.