Coupling Mask R-CNN and Attention Mechanism for Building Extraction and Post-Processing Strategy
Buildings are integral components of urban areas.Extracting buildings from high-resolution remote sensing data holds significant academic importance in areas such as land use analysis,urban planning,and disas-ter risk reduction.For the problems of building extraction,an improved Mask R-CNN building instance seg-mentation model is proposed.Based on the residual neural network fusion convolutional attention model,a re-sidual convolutional attention network is constructed to improve the problem of inadequate feature extraction.The loss function is optimized by adding the Dice Loss method,and then the feature learning process is opti-mized.And a post-processing strategy combining Douglas-Peucker algorithm and Fine polygon regularization algorithm is introduced to make the building contours more regular and smooth.The experimental results show that the improved model improves the detection mAP value by 7.74% at Iou 0.5 and 7.57% at Iou 0.75 com-pared with the original model,and the post-processing strategy improves the F1-Score value by 6.01% com-pared with the original model after selecting the appropriate threshold to optimize the mask.The instance seg-mentation model coupled with Mask R-CNN and attention mechanism improves the small building misdetection and omission problem,building segmentation boundary adhesion problem,and building segmentation accuracy;building post-processing strategy,improves building regularization.
NETL
NSTL
万方数据
