張媛芳
(昆明市日?qǐng)?bào),云南 昆明 650201)
耕地的數(shù)量和質(zhì)量是保持農(nóng)業(yè)可持續(xù)發(fā)展的關(guān)鍵,準(zhǔn)確的耕地分布能夠?yàn)閲覜Q策部門提供重要支撐,利用衛(wèi)星遙感影像可以高精度、高質(zhì)量識(shí)別并提取耕地。 為了解決在遙感圖像中耕地分割與提取的問題,本文通過將原始圖片灰度化并進(jìn)行二值化處理,分離耕地與背景區(qū)域;以主干特征提取網(wǎng)絡(luò)MobileNetV2和加強(qiáng)特征提取結(jié)構(gòu)PSP模塊搭建的PSPNet神經(jīng)網(wǎng)絡(luò)模型進(jìn)行圖片訓(xùn)練,得到訓(xùn)練后的模型,進(jìn)行圖片預(yù)測(cè),從而將圖像中的耕地區(qū)域提取出來。通過改變模型訓(xùn)練的數(shù)據(jù)集和訓(xùn)練次數(shù)的大小,探討這一改變對(duì)圖片提取耕地區(qū)域精準(zhǔn)度的影響。
問題背景:目前高精度的耕地信息提取主要依靠人工解譯,耗費(fèi)大量人力、財(cái)力且效率較低。圖像分割與提取是獲取圖像內(nèi)部信息的兩種重要手段,利用衛(wèi)星遙感影像可以識(shí)別并提取耕地,進(jìn)行自動(dòng)化地塊提取顯得日益重要。
需要解決的問題:建立模型構(gòu)造出分割耕地與背景的方法,可以使其耕地區(qū)域與背景區(qū)域準(zhǔn)確分離出來,并且計(jì)算出耕地區(qū)域在圖像中所占比例;進(jìn)行數(shù)據(jù)集制作,構(gòu)建神經(jīng)網(wǎng)絡(luò)進(jìn)行模型訓(xùn)練,使之通過訓(xùn)練好的模型進(jìn)行預(yù)測(cè),提取出耕地區(qū)域,用labelme軟件進(jìn)行數(shù)據(jù)標(biāo)簽的制作。
問題的研究對(duì)象是遙感圖像,研究內(nèi)容為在遙感圖像地塊中分割耕地與背景,并且通過模型訓(xùn)練,進(jìn)行模型預(yù)測(cè),從遙感圖像中提取出耕地區(qū)域,制作耕地標(biāo)簽圖。問題1要求計(jì)算途中耕地在各圖像中所占比例,問題2要求從給定的測(cè)試圖像中提取出耕地,制作耕地圖標(biāo)簽。
問題1希望能將耕地與背景分離,計(jì)算出耕地在圖像中所占比例。解決問題的關(guān)鍵是得到進(jìn)行二值化圖片的最佳閾值,排除遙感圖像中較深色和較淺色區(qū)域的干擾。首先將原始圖片灰度化,并進(jìn)行二值化處理,耕地區(qū)域的像素點(diǎn)顏色值255(即變?yōu)榘咨?,耕地區(qū)域以外的像素點(diǎn)顏色置0(即變?yōu)楹谏?,?jì)算出白色(耕地)區(qū)域在整個(gè)圖像中所占的比例;問題2構(gòu)建數(shù)據(jù)集,進(jìn)行模型訓(xùn)練,利用模型進(jìn)行圖片預(yù)測(cè),實(shí)現(xiàn)耕地的提取。首先利用已給出的原圖與標(biāo)簽進(jìn)行擴(kuò)充,制作數(shù)據(jù)集,然后利用以mobilenetv2為PSPnet網(wǎng)絡(luò)的主干特征提取網(wǎng)絡(luò),進(jìn)行模型訓(xùn)練,圖片預(yù)測(cè),將耕地區(qū)域從圖片中提取出來,耕地標(biāo)簽圖可在labelme軟件上制作。
(1)假設(shè)耕地邊緣沒有缺陷點(diǎn);
(2)假設(shè)耕地區(qū)域無誤差且不考慮隨機(jī)誤差;
(3)假設(shè)淺色和深色區(qū)域即可看為缺陷點(diǎn);
(4)假設(shè)耕地區(qū)域圖像提取完全。
注:未列出符號(hào)及重復(fù)的符號(hào)以出現(xiàn)處為準(zhǔn)
問題1需要建立一個(gè)能將遙感圖像中的耕地區(qū)域精確分離出來的分割模型,建模時(shí)應(yīng)考慮圖像中待測(cè)區(qū)域之外的邊緣在提取中的干擾。此外,在忽略耕地檢測(cè)邊緣缺陷點(diǎn)以及系統(tǒng)觀測(cè)誤差的基礎(chǔ)上進(jìn)行求解。
模型建立:要精確提取耕地區(qū)域并將區(qū)域外顏色填充為黑色,模型建立過程為:首先將原始圖片灰度化,通過灰度圖像均衡處理,增大對(duì)比度[1],利用大津法得到最佳分割閾值,進(jìn)行二值化處理,白色區(qū)域代表的是耕地類、黑色區(qū)域代表的是背景類,即可得分離后耕地在各圖像中所占比例。
模型求解:將RGB圖像轉(zhuǎn)換為灰度圖像,采用大津法,得到最佳閾值,將灰度圖像轉(zhuǎn)換為二值圖像。設(shè)final_threshold為耕地和背景的分割閾值,灰度級(jí)小于等于final_threshold的為背景,大于final_threshold的為前景(耕地)。前景像素點(diǎn)數(shù)所占的圖像的比例為wo及平均灰度為u0;背景像素點(diǎn)數(shù)所占的圖像的比例為w1,平均灰度為u1。則圖像的總平均灰度為[2]:
前景和背景圖像的方差為:
當(dāng)計(jì)算出來的方差值g取最大值時(shí),可以認(rèn)為前景跟背景的差異達(dá)到最大,此時(shí)的分割閾值final_threshold為最佳閾值。
模型結(jié)果:
(1)Data1圖像經(jīng)二值化處理,如圖1所示。
圖1 Data1 耕地與背景分離圖
(2)Data2圖像經(jīng)二值化處理,如圖2所示。
圖2 Data2 耕地與背景分離圖
(3)Data3圖像經(jīng)二值化處理,如圖3所示。
圖3 Data3 耕地與背景分離圖
(4)Data4圖像經(jīng)二值化處理,如圖4所示。
圖4 Data4 耕地與背景分離圖
(5)Data5圖像經(jīng)二值化處理,如圖5所示。
圖5 Data5 耕地與背景分離圖
(6)Test1圖像經(jīng)二值化處理,如圖6所示。
圖6 Test1 耕地與背景分離圖
計(jì)算出圖片中耕地(白色)區(qū)域在圖像中所占比例,如圖7所示。
圖7 耕地在各圖像中所占比例
問題2要求從測(cè)試圖像中提取耕地區(qū)域,制作耕地標(biāo)簽。處理方法:進(jìn)行數(shù)據(jù)集的制作,在PSPnet網(wǎng)絡(luò)中以MobileNetV2作為主干特征提取網(wǎng)絡(luò),在3x3網(wǎng)絡(luò)結(jié)構(gòu)前利用1x1卷積升維,在3x3網(wǎng)絡(luò)結(jié)構(gòu)后,利用1x1卷積降維,先進(jìn)行擴(kuò)張,再進(jìn)行壓縮[3],加強(qiáng)特征提取結(jié)構(gòu)PSP模塊典型情況下,將輸入進(jìn)來的特征層劃分成6x6,3x3,2x2,1x1的區(qū)域,然后每個(gè)區(qū)域內(nèi)部各自進(jìn)行平均池化[4-5]。利用已經(jīng)搭建好的PSPnet網(wǎng)絡(luò)進(jìn)行圖片訓(xùn)練,得到訓(xùn)練模型,根據(jù)輸入的圖片,提取出圖片中的耕地區(qū)域。
模型建立:完成數(shù)據(jù)集的制作,進(jìn)行模型訓(xùn)練,完成圖片預(yù)測(cè),進(jìn)行耕地的提取。
模型求解:處理已經(jīng)制作好的數(shù)據(jù)集;通過搭建好的Pspnet網(wǎng)絡(luò)進(jìn)行模型訓(xùn)練,得到已經(jīng)訓(xùn)練好的模型;利用已訓(xùn)練好的模型,提取圖片中耕地區(qū)域;用labelme軟件進(jìn)行耕地標(biāo)簽的制作。
模型結(jié)果:
輸入Test1圖像文件,提取耕地區(qū)域,如圖8所示。
圖8 Test1 原圖與耕地提取混合圖
Test1圖像文件的耕地標(biāo)簽,如圖9所示。
圖9 Test1 耕地標(biāo)簽圖
模型訓(xùn)練的數(shù)據(jù)集大小與模型預(yù)測(cè)得到的效果緊密相關(guān)。當(dāng)模型數(shù)據(jù)集過小時(shí),Test1圖片提取耕地區(qū)域圖片如圖10所示。
圖10 Test1 耕地提取圖
當(dāng)數(shù)據(jù)集大小適當(dāng)時(shí),Test1圖片提取耕地區(qū)域圖片如圖8所示。
模型的優(yōu)點(diǎn):分離圖像時(shí),采用大津法尋找最佳閾值使更清晰的看到田埂輪廓。檢測(cè)模型:通過PSPnet神經(jīng)網(wǎng)絡(luò)搭建的模型訓(xùn)練數(shù)據(jù)集,來進(jìn)行圖片預(yù)測(cè),較為精準(zhǔn)的提取出圖片中的耕地區(qū)域。
模型的不足:在進(jìn)行耕地與背景區(qū)域分離時(shí),顏色較深的像素點(diǎn)被置為0(黑色區(qū)域),顏色較淺的像素點(diǎn)被置為255(白色區(qū)域),導(dǎo)致計(jì)算耕地(白色)區(qū)域所占圖像比例不準(zhǔn)確;進(jìn)行模型訓(xùn)練時(shí),數(shù)據(jù)集不能過小,會(huì)導(dǎo)致訓(xùn)練出的模型進(jìn)行預(yù)測(cè)時(shí)不準(zhǔn)確。
依靠人工解譯來提取耕地信息,耗費(fèi)大量人力、財(cái)力且效率較低,基于PSPnet神經(jīng)網(wǎng)絡(luò)的遙感圖像耕地識(shí)別算法研究將對(duì)耕地遙感制圖提供重要幫助,隨著訓(xùn)練數(shù)據(jù)集的增大,進(jìn)行預(yù)測(cè)得到的效果會(huì)大幅度提升,能夠快速、精準(zhǔn)的識(shí)別出田塊。