本發(fā)明屬于圖像處理技術(shù)領(lǐng)域,更進(jìn)一步涉及合成孔徑雷達(dá)SAR圖像目標(biāo)檢測(cè)技術(shù)領(lǐng)域中的一種基于恒虛景CFAR(Constant False Alarm Rate)檢測(cè)與深度學(xué)習(xí)的合成孔徑雷達(dá)SAR(Synthetic Aperture Radar)目標(biāo)檢測(cè)方法。本發(fā)明能夠準(zhǔn)確的檢測(cè)出合成孔徑雷達(dá)SAR圖像的目標(biāo),并且可用于后續(xù)合成孔徑雷達(dá)SAR圖像的目標(biāo)識(shí)別。
背景技術(shù):
合成孔徑雷達(dá)SAR具有全天候、全天時(shí)、分辨率高以及穿透力強(qiáng)等特點(diǎn),廣泛應(yīng)用于軍事偵察和遙感領(lǐng)域。雷達(dá)成像技術(shù)在對(duì)地面目標(biāo)特別是地面靜止目標(biāo)探測(cè)方面具有獨(dú)特的優(yōu)勢(shì),隨著SAR技術(shù)不斷成熟,成像分辨率不斷提高,使得通過(guò)SAR圖像的目標(biāo)識(shí)別技術(shù)受到越來(lái)越廣泛的關(guān)注。
卷積神經(jīng)網(wǎng)絡(luò)是深度學(xué)習(xí)方法的一種,已成為當(dāng)前語(yǔ)音分析和圖像識(shí)別領(lǐng)域的研究熱點(diǎn)。它的權(quán)值共享網(wǎng)絡(luò)結(jié)構(gòu)使之更類(lèi)似于生物神經(jīng)網(wǎng)絡(luò),降低了網(wǎng)絡(luò)模型的復(fù)雜度,減少了權(quán)值的數(shù)量。該優(yōu)點(diǎn)在網(wǎng)絡(luò)的輸入是多維圖像時(shí)表現(xiàn)的更為明顯,使圖像可以直接作為網(wǎng)絡(luò)的輸入,避免了傳統(tǒng)識(shí)別算法中復(fù)雜的特征提取和數(shù)據(jù)重建過(guò)程。卷積網(wǎng)絡(luò)是為識(shí)別二維形狀而特殊設(shè)計(jì)的一個(gè)多層感知器,這種網(wǎng)絡(luò)結(jié)構(gòu)對(duì)平移、比例縮放、傾斜或者共他形式的變形具有高度不變性。
目前,已經(jīng)發(fā)展出很多針對(duì)SAR圖像的目標(biāo)檢測(cè)算法。其中,恒虛警(Constant False Alarm Rate,CFAR)檢測(cè)算法以其簡(jiǎn)單、快速、實(shí)時(shí)性強(qiáng)的特點(diǎn)而被廣泛應(yīng)用于合成孔徑雷達(dá)SAR圖像目標(biāo)檢測(cè)中。此外,根據(jù)不同類(lèi)型的目標(biāo)在SAR圖像上具有不同的表征形式,也相應(yīng)的具有不同的檢測(cè)方法。但這些現(xiàn)有SAR圖像檢測(cè)方法僅利用了SAR圖像局部區(qū)域的統(tǒng)計(jì)特性,僅能做到像素級(jí)別的檢測(cè),不能做到端對(duì)端檢測(cè),在復(fù)雜場(chǎng)景下在復(fù)雜場(chǎng)景下,檢測(cè)性能較差檢測(cè)性能較差。
中國(guó)人民解放軍國(guó)防科學(xué)技術(shù)大學(xué)其申請(qǐng)的專(zhuān)利“一種SAR圖像雙邊CFAR艦船目標(biāo)檢測(cè)方法”(專(zhuān)利申請(qǐng)?zhí)?01510011762.6,公布號(hào)CN104537675A)中公開(kāi)了一種SAR圖像雙邊CFAR艦船目標(biāo)檢測(cè)方法。該方法將SAR圖像中的每個(gè)像素點(diǎn)的亮度值和空間值聯(lián)合,利用恒虛景CFAR檢測(cè)方法對(duì)聯(lián)合圖像進(jìn)行檢測(cè),得到艦船目標(biāo)檢測(cè)結(jié)果。該方法存在的不足之處是,對(duì)切片的處理是像素級(jí)別的處理,不能做到圖片級(jí)別的檢測(cè),檢測(cè)門(mén)限的設(shè)置不方便。
西安電子科技大學(xué)在其申請(qǐng)的專(zhuān)利“基于CFAR和稀疏表示的高分辨SAR艦船檢測(cè)方法”(專(zhuān)利申請(qǐng)?zhí)?01310280179.6,公布號(hào)CN103400156A)中公開(kāi)了一種基于CFAR和稀疏表示的高分辨SAR艦船檢測(cè)方法。該方法在正常恒虛景CFAR檢測(cè)的基礎(chǔ)上,對(duì)切片提取特征向量,并通過(guò)稀疏表示分類(lèi)器做鑒別,得到最終的艦船檢測(cè)結(jié)果。該方法的不足之處是檢測(cè)過(guò)程要經(jīng)過(guò)檢測(cè)、鑒別兩個(gè)步驟,不能做到端對(duì)端檢測(cè),并且在復(fù)雜場(chǎng)景下,檢測(cè)性能較差。
技術(shù)實(shí)現(xiàn)要素:
本發(fā)明的目的在于克服上述現(xiàn)有技術(shù)的不足,提出一種基于CFAR檢測(cè)與深度學(xué)習(xí)的SAR目標(biāo)檢測(cè)方法。本發(fā)明克服了現(xiàn)有技術(shù)中的合成孔徑雷達(dá)SAR圖像目標(biāo)檢測(cè)方法中僅利用了SAR圖像局部區(qū)域的統(tǒng)計(jì)特性,僅能做到像素級(jí)別的檢測(cè)的問(wèn)題,同時(shí)做到了端對(duì)端檢測(cè),提高了復(fù)雜場(chǎng)景下在復(fù)雜場(chǎng)景下,如:強(qiáng)雜波較多,檢測(cè)性能較差合成孔徑雷達(dá)SAR圖像目標(biāo)檢測(cè)定位的準(zhǔn)確性。
為實(shí)現(xiàn)上述目的,本發(fā)明具體實(shí)現(xiàn)步驟包括如下:
(1)獲取SAR圖像:
(1a)從MiniSAR數(shù)據(jù)集中隨機(jī)選取100幅SAR圖像;
(1b)從MiniSAR數(shù)據(jù)集中選取與所選SAR圖像對(duì)應(yīng)的目標(biāo)坐標(biāo)信息和類(lèi)別標(biāo)簽;
(1c)將所選取的SAR圖像以及目標(biāo)坐標(biāo)信息和類(lèi)別標(biāo)簽組成訓(xùn)練樣本集;
(2)擴(kuò)充訓(xùn)練樣本集:
對(duì)訓(xùn)練樣本集中每幅SAR圖像的待識(shí)別目標(biāo)區(qū)域進(jìn)行100次的隨機(jī)平移,將每次平移后的訓(xùn)練樣本圖像組成擴(kuò)充后的訓(xùn)練樣本集;
(3)構(gòu)建Faster-RCNN模型的網(wǎng)絡(luò)結(jié)構(gòu):
(3a)構(gòu)建一個(gè)三層卷積網(wǎng)絡(luò),其中,第一層為卷積層,使用96個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為4個(gè)像素,輸出96個(gè)特征圖,每個(gè)特征圖經(jīng)過(guò)一個(gè)下采樣進(jìn)行降維,下采樣的核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為2個(gè)像素,將降維后的特征圖輸入到第二層;第二層為卷積層,使用96個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為2個(gè)像素,輸出96個(gè)特征圖,每個(gè)特征圖經(jīng)過(guò)一個(gè)下采樣進(jìn)行降維,下采樣的核窗口大小為個(gè)像素3*3,相鄰局部接受域的中心距離為2個(gè)像素,將降維后的特征圖輸入到第三層;第三層為卷積層,使用256個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為2個(gè)像素,輸出256個(gè)特征圖;
(3b)在步驟(3a)的三層卷積網(wǎng)絡(luò)后引入第四層卷積層,該層使用256個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為1個(gè)像素,輸出256個(gè)特征圖,特征圖輸入到第五、六層,與此同時(shí),對(duì)于每個(gè)滑窗,以滑窗中心為中心,分別以1282、2562、5122三種尺寸,2:1,1:1,1:2三種長(zhǎng)寬比構(gòu)建9個(gè)anchor boxes;第五、六層為同級(jí)的全連接層,第五層為18個(gè)神經(jīng)元的全連接,第六層為36個(gè)神經(jīng)元的全連接層,將以上各層合在一起,得到用于提取圖像的感興趣區(qū)域RPN網(wǎng)絡(luò);
(3c)在步驟(3a)的三層卷積后引入第四層RoI池化層,第4層為核窗口大小自適應(yīng)于輸出6*6個(gè)像素特征圖,相鄰局部接受域的中心距離為2個(gè)像素的下采樣層,經(jīng)下采樣得到降維后的特征圖輸入到第五層;第五層為有1000個(gè)神經(jīng)元的全連接層,將第五層輸出輸入到第六、七層;第六、七層為同級(jí)的全連接層,第六層為2個(gè)神經(jīng)元的全連接層,第七層為8個(gè)神經(jīng)元的全連接層,將以上各層合在一起,得到用于圖像的檢測(cè)和分類(lèi)Fast-RCNN網(wǎng)絡(luò);
(4)獲得訓(xùn)練好的RPN模型:
(4a)將擴(kuò)充后的訓(xùn)練樣本集輸入到RPN網(wǎng)絡(luò),利用隨機(jī)梯度下降算法,分別計(jì)算RPN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值;
(4b)利用反向傳播算法,計(jì)算RPN網(wǎng)絡(luò)中每一層的誤差靈敏度,用每一層的誤差靈敏度更新該層的權(quán)值;
(4c)判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到訓(xùn)練好的RPN網(wǎng)絡(luò)模型,執(zhí)行步驟(5),否則,執(zhí)行步驟(4a);
(5)獲得訓(xùn)練好的Fast-RCNN模型:
(5a)將擴(kuò)充后的訓(xùn)練樣本集以及訓(xùn)練好的RPN網(wǎng)絡(luò)模型的anchor boxes輸入到Fast-RCNN網(wǎng)絡(luò)中,利用隨機(jī)梯度下降算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值;
(5b)利用反向傳播算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)中每一層的誤差靈敏度,用每一層的誤差靈敏度更新該層的權(quán)值;
(5c)判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到訓(xùn)練好的Fast-RCNN網(wǎng)絡(luò)模型,執(zhí)行步驟(6),否則,執(zhí)行步驟(5a);
(6)獲得微調(diào)的RPN網(wǎng)絡(luò):
(6a)將訓(xùn)練好的Fast-RCNN網(wǎng)絡(luò)模型的前三層的參數(shù)固定不變,將擴(kuò)充后的訓(xùn)練樣本集輸入到RPN網(wǎng)絡(luò)中;
(6b)利用隨機(jī)梯度下降算法,計(jì)算RPN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值;
(6c)利用反向傳播算法,計(jì)算RPN網(wǎng)絡(luò)第四、五、六層的誤差靈敏度,用第四、五、六層的誤差靈敏度更新RPN網(wǎng)絡(luò)的第四、五、六層的權(quán)值;
(6d)判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到微調(diào)的RPN網(wǎng)絡(luò)模型,執(zhí)行步驟(7),否則,執(zhí)行步驟(6a);
(7)獲得訓(xùn)練好的Faster-RCNN模型:
(7a)將訓(xùn)練好的Fast-RCNN網(wǎng)絡(luò)模型的前三層的參數(shù)固定不變,將擴(kuò)充后的訓(xùn)練樣本集以及微調(diào)的RPN網(wǎng)絡(luò)模型的anchor boxes輸入到Fast-RCNN網(wǎng)絡(luò)中;
(7b)利用隨機(jī)梯度下降算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值;
(7c)利用反向傳播算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)第四、五、六、七層的誤差靈敏度,用第四、五、六、七層的誤差靈敏度更新Fast-RCNN網(wǎng)絡(luò)第四、五、六、七層的權(quán)值;
(7d)判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到訓(xùn)練好的Faster-RCNN網(wǎng)絡(luò)的各層網(wǎng)絡(luò)參數(shù),執(zhí)行步驟(8),否則,執(zhí)行步驟(7a);
(8)目標(biāo)檢測(cè):
(8a)對(duì)測(cè)試數(shù)據(jù)進(jìn)行恒虛景CFAR檢測(cè),以恒虛景CFAR二值檢測(cè)結(jié)果的各個(gè)目標(biāo)質(zhì)心為中心,提取與各個(gè)目標(biāo)質(zhì)心對(duì)應(yīng)的檢測(cè)切片;
(8b)用訓(xùn)練好的Faster-RCNN模型對(duì)各個(gè)檢測(cè)切片進(jìn)行目標(biāo)檢測(cè),得到各個(gè)切片的目標(biāo)檢測(cè)結(jié)果;
(8c)將各個(gè)切片的目標(biāo)檢測(cè)結(jié)果,按照其在原始SAR圖像的相對(duì)位置進(jìn)行合并,得到測(cè)試數(shù)據(jù)的目標(biāo)檢測(cè)結(jié)果。
本發(fā)明與現(xiàn)有技術(shù)相比具有如下優(yōu)點(diǎn):
第一,由于本發(fā)明利用Faster-RCNN模型進(jìn)行目標(biāo)檢測(cè),克服了現(xiàn)有技術(shù)對(duì)切片的處理是像素級(jí)別的處理,不能做到圖片級(jí)別的檢測(cè),檢測(cè)門(mén)限的設(shè)置不方便的問(wèn)題,使本發(fā)明具有能夠完成圖像級(jí)檢測(cè),得到目標(biāo)區(qū)域的檢測(cè)概率,方便后續(xù)的檢測(cè)門(mén)限調(diào)整的優(yōu)點(diǎn)。
第二,由于本發(fā)明先利用恒虛景CFAR檢測(cè)方法進(jìn)行切片,然后利用Faster-RCNN模型對(duì)切片進(jìn)行目標(biāo)檢測(cè),克服了現(xiàn)有技術(shù)檢測(cè)過(guò)程要經(jīng)過(guò)檢測(cè)、鑒別兩個(gè)步驟,不能做到端對(duì)端檢測(cè),耗時(shí)較長(zhǎng)的問(wèn)題,使本發(fā)明具有能夠完成端對(duì)端檢測(cè),將檢測(cè)與鑒別集成于一體的優(yōu)點(diǎn)。
第三,由于本發(fā)明利用訓(xùn)練好的Faster-RCNN模型的各層卷積網(wǎng)絡(luò)參數(shù),挖掘SAR圖像目標(biāo)的一些高層特征,克服了現(xiàn)有技術(shù)復(fù)雜場(chǎng)景下在復(fù)雜場(chǎng)景下,檢測(cè)性能較差的問(wèn)題,使本發(fā)明具有在復(fù)雜場(chǎng)景下檢測(cè)性能較好的優(yōu)點(diǎn)。
附圖說(shuō)明
圖1是本發(fā)明的流程圖;
圖2是本發(fā)明仿真實(shí)驗(yàn)中輸入的合成孔徑雷達(dá)SAR圖;
圖3是本發(fā)明仿真實(shí)驗(yàn)結(jié)果圖。
具體實(shí)施方式
下面結(jié)合附圖對(duì)本發(fā)明做進(jìn)一步的描述。
參照附圖1,本發(fā)明的具體步驟如下。
步驟1,獲取SAR圖像。
從MiniSAR數(shù)據(jù)集中隨機(jī)選取100幅SAR圖像。
從MiniSAR數(shù)據(jù)集中選取與所選SAR圖像對(duì)應(yīng)的目標(biāo)坐標(biāo)信息和類(lèi)別標(biāo)簽。
將所選取的SAR圖像以及目標(biāo)坐標(biāo)信息和類(lèi)別標(biāo)簽組成訓(xùn)練樣本集。
步驟2,擴(kuò)充訓(xùn)練樣本集。
對(duì)訓(xùn)練樣本集中每幅SAR圖像的待識(shí)別目標(biāo)區(qū)域進(jìn)行100次的隨機(jī)平移,將每次平移后的訓(xùn)練樣本圖像組成擴(kuò)充后的訓(xùn)練樣本集。
第1步,在matlab中讀取訓(xùn)練樣本集中的每一個(gè)SAR圖像,得到對(duì)應(yīng)每一個(gè)SAR圖像的二維坐標(biāo)系、以及該SAR圖像中每個(gè)像素點(diǎn)在此坐標(biāo)系下的坐標(biāo)(x,y);
第2步,采用圍繞邊界方法,對(duì)訓(xùn)練樣本集中每一個(gè)SAR圖像的背景,進(jìn)行鏡像反射填充,得到填充后的SAR圖像;
第3步,將填充后SAR圖像中的任一像素點(diǎn)的坐標(biāo)(x,y)沿x軸平移tx個(gè)單位,沿y軸平移ty個(gè)單位,得到(x',y'),其中x'=x+tx,y'=y(tǒng)+ty,tx為在正負(fù)二分之待識(shí)別目標(biāo)區(qū)域的長(zhǎng)的區(qū)間中選取的任一整數(shù),ty為在正負(fù)二分之待識(shí)別目標(biāo)區(qū)域的寬的區(qū)間中選取的任一整數(shù);
第4步,以第1步確定的坐標(biāo)系的橫坐標(biāo)為二分之待識(shí)別目標(biāo)區(qū)域的長(zhǎng),縱坐標(biāo)為二分之待識(shí)別目標(biāo)區(qū)域的寬的坐標(biāo)為起點(diǎn)坐標(biāo),從填充、平移后的SAR圖像中截取與原始SAR圖像訓(xùn)練樣本同樣大小的區(qū)域,得到一個(gè)新的訓(xùn)練樣本圖像;
第5步,判斷平移次數(shù)是否達(dá)到100次,若是,則執(zhí)行第6步,否則,執(zhí)行第1步;
第6步,在隨機(jī)平移得到的所有新訓(xùn)練樣本圖像上,標(biāo)記與原始訓(xùn)練樣本圖像相同的類(lèi)別標(biāo)簽。
第7步,用所有標(biāo)有類(lèi)別標(biāo)簽的新樣本圖像與原始圖像,組成擴(kuò)充后的訓(xùn)練樣本集。
步驟3,構(gòu)建Faster-RCNN模型的網(wǎng)絡(luò)結(jié)構(gòu)。
Faster-RCNN模型網(wǎng)絡(luò)結(jié)構(gòu)的構(gòu)建方法參見(jiàn)Shaoqing Ren等人于2015發(fā)表在NIPS上的文章《Faster R-CNN:Towards Real-Time Object Detection with Region Proposal Networks》。Faster-RCNN模型的網(wǎng)絡(luò)結(jié)構(gòu)包括RPN網(wǎng)絡(luò)跟Fast-RCNN網(wǎng)絡(luò)兩部分,具體的構(gòu)建方法是:
構(gòu)建一個(gè)三層卷積網(wǎng)絡(luò),其中,第一層為卷積層,使用96個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為4個(gè)像素,輸出96個(gè)特征圖,每個(gè)特征圖經(jīng)過(guò)一個(gè)下采樣進(jìn)行降維,下采樣的核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為2個(gè)像素,將降維后的特征圖輸入到第二層;第二層為卷積層,使用96個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為2個(gè)像素,輸出96個(gè)特征圖,每個(gè)特征圖經(jīng)過(guò)一個(gè)下采樣進(jìn)行降維,下采樣的核窗口大小為個(gè)像素3*3,相鄰局部接受域的中心距離為2個(gè)像素,將降維后的特征圖輸入到第三層;第三層為卷積層,使用256個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為2個(gè)像素,輸出256個(gè)特征圖。
構(gòu)建一個(gè)用于提取圖像的感興趣區(qū)域RPN網(wǎng)絡(luò),在第1步的三層卷積網(wǎng)絡(luò)后引入第四層卷積層,該層使用256個(gè)卷積核,卷積核窗口大小為3*3個(gè)像素,相鄰局部接受域的中心距離為1個(gè)像素,輸出256個(gè)特征圖,特征圖輸入到第五、六層,與此同時(shí),對(duì)于每個(gè)滑窗,以滑窗中心為中心,分別以1282、2562、5122三種尺寸,2:1,1:1,1:2三種長(zhǎng)寬比構(gòu)建9個(gè)anchor boxes;第五、六層為同級(jí)的全連接層,第五層為18個(gè)神經(jīng)元的全連接,第六層為36個(gè)神經(jīng)元的全連接層。
構(gòu)建用于圖像的檢測(cè)和分類(lèi)Fast-RCNN網(wǎng)絡(luò),在第1步的三層卷積后引入第四層RoI池化層,第4層為核窗口大小自適應(yīng)于輸出6*6個(gè)像素特征圖,相鄰局部接受域的中心距離為2個(gè)像素的下采樣層,經(jīng)下采樣得到降維后的特征圖輸入到第五層;第五層為有1000個(gè)神經(jīng)元的全連接層,將第五層輸出輸入到第六、七層;第六、七層為同級(jí)的全連接層,第六層為2個(gè)神經(jīng)元的全連接層,第七層為8個(gè)神經(jīng)元的全連接層。
步驟4,獲得訓(xùn)練好的RPN模型。
第1步,將擴(kuò)充后的訓(xùn)練樣本集輸入到RPN網(wǎng)絡(luò),利用隨機(jī)梯度下降算法,分別計(jì)算RPN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值。
第2步,利用反向傳播算法,計(jì)算RPN網(wǎng)絡(luò)中每一層的誤差靈敏度,用每一層的誤差靈敏度更新該層的權(quán)值。
第3步,判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到訓(xùn)練好的RPN網(wǎng)絡(luò)模型,執(zhí)行步驟5,否則,執(zhí)行第1步。
步驟5,獲得訓(xùn)練好的Fast-RCNN模型。
第1步,將擴(kuò)充后的訓(xùn)練樣本集以及訓(xùn)練好的RPN網(wǎng)絡(luò)模型的anchor boxes輸入到Fast-RCNN網(wǎng)絡(luò)中,利用隨機(jī)梯度下降算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值。
第2步,利用反向傳播算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)中每一層的誤差靈敏度,用每一層的誤差靈敏度更新該層的權(quán)值。
第3步,判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到訓(xùn)練好的Fast-RCNN網(wǎng)絡(luò)模型,執(zhí)行步驟6,否則,執(zhí)行第1步。
步驟6,獲得微調(diào)的RPN網(wǎng)絡(luò)。
第1步,將訓(xùn)練好的Fast-RCNN網(wǎng)絡(luò)模型的前三層的參數(shù)固定不變,將擴(kuò)充后的訓(xùn)練樣本集輸入到RPN網(wǎng)絡(luò)中。
第2步,利用隨機(jī)梯度下降算法,計(jì)算RPN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值。
第3步,利用反向傳播算法,計(jì)算RPN網(wǎng)絡(luò)第四、五、六層的誤差靈敏度,用第四、五、六層的誤差靈敏度更新RPN網(wǎng)絡(luò)的第四、五、六層的權(quán)值。
第4步,判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到微調(diào)的RPN網(wǎng)絡(luò)模型,執(zhí)行步驟7,否則,執(zhí)行第1步。
步驟7,獲得訓(xùn)練好的Faster-RCNN模型。
第1步,將訓(xùn)練好的Fast-RCNN網(wǎng)絡(luò)模型的前三層的參數(shù)固定不變,將擴(kuò)充后的訓(xùn)練樣本集以及微調(diào)的RPN網(wǎng)絡(luò)模型的anchor boxes輸入到Fast-RCNN網(wǎng)絡(luò)中。
第2步,利用隨機(jī)梯度下降算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)中每一層的代價(jià)函數(shù)的梯度值,用每一層的梯度值更新該層的權(quán)值。
第3步,利用反向傳播算法,計(jì)算Fast-RCNN網(wǎng)絡(luò)第四、五、六、七層的誤差靈敏度,用第四、五、六、七層的誤差靈敏度更新Fast-RCNN網(wǎng)絡(luò)第四、五、六、七層的權(quán)值。
第4步,判斷更新前的權(quán)值與更新后的權(quán)值是否相等,若是,則網(wǎng)絡(luò)訓(xùn)練完畢,得到訓(xùn)練好的Faster-RCNN網(wǎng)絡(luò)的各層網(wǎng)絡(luò)參數(shù),執(zhí)行步驟8,否則,執(zhí)行第1步。
步驟8,目標(biāo)檢測(cè)。
對(duì)測(cè)試數(shù)據(jù)進(jìn)行恒虛景CFAR檢測(cè),以恒虛景CFAR二值檢測(cè)結(jié)果的各個(gè)目標(biāo)質(zhì)心為中心,提取與各個(gè)目標(biāo)質(zhì)心對(duì)應(yīng)的檢測(cè)切片。具體做法是:
第1步,對(duì)輸入的SAR圖像進(jìn)行滑窗處理,得到一個(gè)對(duì)應(yīng)的中心像素點(diǎn),滑窗內(nèi)部為保護(hù)區(qū)域,滑窗周?chē)鸀殡s波區(qū)域;
第2步,按照下式,計(jì)算每次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域像素的均值和方差:
其中,μi表示第i次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域像素的均值,∑表示求和操作,N表示每次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域像素的個(gè)數(shù),j表示每次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域中的第j個(gè)像素,xj表示每次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域中的第j個(gè)像素的強(qiáng)度,σi表示第i次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域像素的方差,表示開(kāi)方操作;
第3步,按照下式,計(jì)算每次滑窗的中心像素點(diǎn)的檢測(cè)統(tǒng)計(jì)量:
其中,Di表示第i次滑窗的中心像素點(diǎn)的檢測(cè)統(tǒng)計(jì)量,Xi表示第i次滑窗的中心像素點(diǎn)的強(qiáng)度,μi表示第i次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域像素的均值,σi表示第i次滑窗的中心像素點(diǎn)周?chē)s波區(qū)域像素的方差;
第4步,在[0,0.2]的范圍內(nèi),隨機(jī)設(shè)定一個(gè)檢測(cè)閾值;
第5步,判斷檢測(cè)統(tǒng)計(jì)量是否大于檢測(cè)閾值,若是,將該像素點(diǎn)作為目標(biāo)點(diǎn),否則,將該像素點(diǎn)作為雜波點(diǎn);
第6步,判斷是否遍歷輸入的SAR圖像的所有的中心像素點(diǎn),若是,則得到恒虛景CFAR二值檢測(cè)結(jié)果,否則,執(zhí)行第1步。
第7步,將所有的目標(biāo)點(diǎn)合并為區(qū)域目標(biāo);
第8步,以區(qū)域目標(biāo)的質(zhì)心為中心,用矩形框框出,得到目標(biāo)質(zhì)心對(duì)應(yīng)的檢測(cè)切片。
用訓(xùn)練好的Faster-RCNN模型對(duì)各個(gè)檢測(cè)切片進(jìn)行目標(biāo)檢測(cè),得到各個(gè)切片的目標(biāo)檢測(cè)結(jié)果。具體做法是:
第1步,將恒虛景CFAR檢測(cè)后所得的切片輸入到訓(xùn)練好的Faster-RCNN模型中,每一個(gè)切片中的每一個(gè)anchor box得到一個(gè)概率值;
第2步,在[0.6,1]的范圍內(nèi),隨機(jī)選取一個(gè)概率值作為閾值;
第3步,判斷anchor box的概率值是否大于閾值,若是,將該anchor box作為目標(biāo),否則,將該anchor box作為非目標(biāo)。
第4步,判斷是否遍歷輸入的所有切片,若是,則得到測(cè)試數(shù)據(jù)的目標(biāo)檢測(cè)結(jié)果,否則,執(zhí)行第1步。
將各個(gè)切片的目標(biāo)檢測(cè)結(jié)果按在原始SAR圖像的位置進(jìn)行合并,得到整個(gè)測(cè)試數(shù)據(jù)的目標(biāo)檢測(cè)結(jié)果。
下面結(jié)合仿真圖對(duì)本發(fā)明做進(jìn)一步的描述。
1.仿真條件:
本發(fā)明的仿真實(shí)驗(yàn)中原始訓(xùn)練樣本集為MiniSAR數(shù)據(jù)集中隨機(jī)選取的100幅SAR圖像,測(cè)試樣本為一副場(chǎng)景SAR圖像,如附圖2所示。圖像的大小為2510×1638,其中包含了車(chē)輛類(lèi)人工目標(biāo),也包含了樹(shù)木、田野等自然目標(biāo),仿真實(shí)驗(yàn)的目的是檢測(cè)和定位出圖像中所有類(lèi)型的人工目標(biāo)。
本發(fā)明的仿真實(shí)驗(yàn)的計(jì)算機(jī)環(huán)境:操作系統(tǒng)為L(zhǎng)inux 14.04版本,處理器為Intel(R)Core(TM)i7-4790k處理器的主頻率為4.00GHz;軟件平臺(tái)為:MatlabR2012a、caffe。
2.仿真內(nèi)容與結(jié)果分析:
本發(fā)明的仿真實(shí)驗(yàn)是對(duì)合成孔徑雷達(dá)SAR圖像中的人工目標(biāo)進(jìn)行檢測(cè),首先,利用原始訓(xùn)練樣本集中的100幅SAR圖像對(duì)Faster‐RCNN模型進(jìn)行訓(xùn)練,得到訓(xùn)練好的模型。
然后,將測(cè)試樣本輸入到訓(xùn)練好的Faster‐RCNN模型中,分別以0.6,0.7,0.8的概率閾值進(jìn)行篩選檢測(cè),檢測(cè)完畢后統(tǒng)計(jì)檢測(cè)率與虛景率對(duì)檢測(cè)結(jié)果進(jìn)行量化,其中,
附圖3是本發(fā)明仿真實(shí)驗(yàn)結(jié)果,反映了恒虛景CFAR檢測(cè)與本發(fā)明的性能比較,其中圖3中的橫坐標(biāo)表示虛景率,縱坐標(biāo)表示檢測(cè)率,帶三角的折線表示恒虛景CFAR檢測(cè)的性能折線,帶圓形的折線表示本發(fā)明在概率閾值0.6下的性能折線,帶菱形的折線表示本發(fā)明在概率閾值0.7下的性能折線,帶正方形的折線表示本發(fā)明在概率閾值0.6下的性能折線。
通過(guò)附圖3性能比較結(jié)果可以看到,本發(fā)明能夠正確的檢測(cè)出合成孔徑雷達(dá)SAR圖像中人工目標(biāo),且在復(fù)雜場(chǎng)景下檢測(cè)性能較好,虛景率低,比恒虛景CFAR檢測(cè)有明顯的提高,使用本發(fā)明方法對(duì)合成孔徑雷達(dá)SAR圖像進(jìn)行人工目標(biāo)檢測(cè)能夠獲得較精確的結(jié)果。