摘 要: Mask RCNN是当前最高水平的实例分割算法,本文将该算法应用到高分辨率遥感图像建筑物提取中, 提出了一种高效、准确的高分辨率遥感图像建筑物提取方法。首先,利用Tensorflow和Keras深度学习框架搭建Mask RCNN网络模型;然后,通过有监督学习方式在IAILD数据集上进行模型学习。利用训练出的模型对测试集进行建筑 物提取实验,通过与基于KNN和SVM等建筑物提取方法对比可以看出,本文方法可以更加完整的、准确的提取出建筑 物。采用mAP评价指标对实验结果进行定量评价,本文算法的查全率和查准率均大于对比算法,且多次实验中本文算 法的mAP均在81%以上,验证了基于卷积网络的高分辨率遥感图像建筑物提取的有效性和准确性。 |
关键词: 深度学习;建筑物提取;Mask RCNN;卷积网络 |
中图分类号: TP183
文献标识码: A
|
基金项目: 基于深度学习的遥感图像识别与目标检测(项目编号:CXZZSS2018119). |
|
A Study of Building Extraction from Remote Sensing Imagery Based on Convolution Network |
FU Fa,WEI Jianying,ZHANG Lina
|
( School of Information Engineering, Hebei GEO University, Shijiazhuang 050000, China)
|
Abstract: Mask RCNN,the state-of-the-art instance segmentation algorithm,is adopted in the building extraction from high-resolution remote sensing imagery.The paper proposes an efficient and accurate building extraction method from high-resolution remote sensing imagery.Firstly,Tensorflow and Keras deep learning framework are used to build the Mask RCNN network model;then,the model learning is carried out on the IAILD data set by supervised learning.By using the trained model,the experiment of building extraction is carried out on the test set.By comparing with the building extraction methods,such as KNN and SVM,it can be seen that this method can be more complete and accurate to extract the building. The mAP evaluation index is used to evaluate the experimental results quantitatively.The recall and precision of the algorithm are all larger than the contrast algorithms,and the mAP of the algorithm is more than 81% in this paper,which verifies the validity and accuracy of the high resolution remote sensing image building extraction based on convolution neural network. |
Keywords: deep learning;building extraction,Mask RCNN;CNN |