由于遙感影像中建筑物種類繁多且與周圍環(huán)境信息混淆,傳統(tǒng)方法難以實現(xiàn)建筑物的準(zhǔn)確高效提取.本文提出了一種基于改進Mask-RCNN的建筑物自動提取方法,利用PyTorch深度學(xué)習(xí)框架搭建改進Mask-RCNN網(wǎng)絡(luò)模型架構(gòu),在網(wǎng)絡(luò)的設(shè)計中添加了路徑聚合網(wǎng)絡(luò)和特征增強功能,通過監(jiān)督和遷移學(xué)習(xí)的方式對Inria航空影像標(biāo)簽數(shù)據(jù)集進行多線程迭代訓(xùn)練與模型優(yōu)化學(xué)習(xí),實現(xiàn)了建筑物的自動精確分割和提取.基于不同開源數(shù)據(jù)集,分別與SVM、FCN、U-net和Mask-RCNN等建筑物提取算法進行對比,實驗表明,本文方法可以高效準(zhǔn)確、高效地提取建筑物,對于同一個數(shù)據(jù)集,提取結(jié)果的mAP、mRecall、mPrecision和F1分數(shù)這4個評價指標(biāo)均優(yōu)于對比算法. 

【文章來源】：計算機系統(tǒng)應(yīng)用. 2020,29(09)

【文章頁數(shù)】：8 頁


本文編號：2970195