基于YOLO-v3模型壓縮的衛(wèi)星圖像船只實(shí)時(shí)檢測
發(fā)布時(shí)間:2022-01-22 06:23
常見的目標(biāo)檢測模型由于模型參數(shù)量較大,往往難以部署在無人機(jī)、衛(wèi)星等移動(dòng)嵌入式設(shè)備上。為了對(duì)船只進(jìn)行實(shí)時(shí)監(jiān)測,將目標(biāo)檢測模型部署在計(jì)算能力較弱的設(shè)備上,對(duì)基于計(jì)算機(jī)視覺的衛(wèi)星圖像船只目標(biāo)檢測方法進(jìn)行研究。針對(duì)衛(wèi)星圖像中船艦的形狀長寬比例特點(diǎn),采用K-means++聚類算法選取初始的錨點(diǎn)框;接著對(duì)模型進(jìn)行多尺度訓(xùn)練,將多尺度金字塔圖像作為模型訓(xùn)練的輸入;將YOLO-v3目標(biāo)檢測算法的批歸一化層的尺度因子作為通道重要性的度量指標(biāo),對(duì)YOLO-v3模型進(jìn)行剪枝壓縮。實(shí)驗(yàn)結(jié)果表明,采用的模型剪枝和壓縮方法能有效地對(duì)模型進(jìn)行壓縮,模型的參數(shù)量減少了91.5%,模型檢測時(shí)間縮短了60%,極大地減少了系統(tǒng)計(jì)算性能的開銷。當(dāng)采用的初始錨點(diǎn)框個(gè)數(shù)為6個(gè)時(shí),平均準(zhǔn)確率(mAP)達(dá)到77.31%,滿足了衛(wèi)星圖像船艦實(shí)時(shí)性檢測的需求。
【文章來源】:液晶與顯示. 2020,35(11)北大核心CSCD
【文章頁數(shù)】:9 頁
【部分圖文】:
多尺度圖像金字塔
YOLO-v3算法將目標(biāo)區(qū)域檢測與物體類別預(yù)測看作回歸問題。YOLO-v3算法的特征提取網(wǎng)絡(luò)為Darknet53。Darknet53包含52個(gè)卷積層,用于對(duì)圖像進(jìn)行特征提取,為了防止深層網(wǎng)絡(luò)訓(xùn)練過程中發(fā)生模型退化,YOLO-v3借鑒了ResNet[12]的做法,在層與層之間進(jìn)行了鏈路的連接。最后一層為1個(gè)全連接層。YOLO-v3通過單個(gè)神經(jīng)網(wǎng)絡(luò),即可實(shí)現(xiàn)對(duì)圖像中目標(biāo)的定位和分類。Darknet-53網(wǎng)絡(luò)結(jié)構(gòu)如圖2所示。Darknet-53網(wǎng)絡(luò)對(duì)圖像進(jìn)行了5次下采樣,每次采樣的步長為2。其中卷積模塊的結(jié)構(gòu)如圖3。
YOLO-v3算法假設(shè)目標(biāo)散落在圖像中各個(gè)區(qū)域,因此將圖像劃分為多個(gè)子區(qū)域。接著對(duì)各個(gè)區(qū)域預(yù)測多個(gè)錨點(diǎn)框(Bounding Box),并求出每個(gè)邊框的置信度(Confidence)。置信度分?jǐn)?shù)的大小,反映了錨點(diǎn)框內(nèi)存在目標(biāo)物體的準(zhǔn)確程度。置信度定義如下式:式中,Pr(Object)表示錨點(diǎn)框中存在目標(biāo)圖像的概率大小。IoU表示預(yù)測的目標(biāo)框與真實(shí)的標(biāo)注框之間的交并比,IoU的計(jì)算公式如下:
【參考文獻(xiàn)】:
期刊論文
[1]艦船目標(biāo)識(shí)別技術(shù)研究進(jìn)展[J]. 馬嘯,邵利民,金鑫,徐冠雷. 科技導(dǎo)報(bào). 2019(24)
[2]無人機(jī)海監(jiān)測繪技術(shù)應(yīng)用下艦船遙感圖像目標(biāo)檢測[J]. 安潔玉,丁斌芬. 艦船科學(xué)技術(shù). 2019(24)
[3]改進(jìn)的YOLO模型及其在艦船目標(biāo)識(shí)別中的應(yīng)用[J]. 馬嘯,邵利民,金鑫,徐冠雷. 電訊技術(shù). 2019(08)
[4]基于YOLO和嵌入式系統(tǒng)的車流量檢測[J]. 馬永杰,宋曉鳳. 液晶與顯示. 2019(06)
[5]基于Faster-RCNN和多分辨率SAR的海上艦船目標(biāo)檢測[J]. 胡炎,單子力,高峰. 無線電工程. 2018(02)
本文編號(hào):3601700
【文章來源】:液晶與顯示. 2020,35(11)北大核心CSCD
【文章頁數(shù)】:9 頁
【部分圖文】:
多尺度圖像金字塔
YOLO-v3算法將目標(biāo)區(qū)域檢測與物體類別預(yù)測看作回歸問題。YOLO-v3算法的特征提取網(wǎng)絡(luò)為Darknet53。Darknet53包含52個(gè)卷積層,用于對(duì)圖像進(jìn)行特征提取,為了防止深層網(wǎng)絡(luò)訓(xùn)練過程中發(fā)生模型退化,YOLO-v3借鑒了ResNet[12]的做法,在層與層之間進(jìn)行了鏈路的連接。最后一層為1個(gè)全連接層。YOLO-v3通過單個(gè)神經(jīng)網(wǎng)絡(luò),即可實(shí)現(xiàn)對(duì)圖像中目標(biāo)的定位和分類。Darknet-53網(wǎng)絡(luò)結(jié)構(gòu)如圖2所示。Darknet-53網(wǎng)絡(luò)對(duì)圖像進(jìn)行了5次下采樣,每次采樣的步長為2。其中卷積模塊的結(jié)構(gòu)如圖3。
YOLO-v3算法假設(shè)目標(biāo)散落在圖像中各個(gè)區(qū)域,因此將圖像劃分為多個(gè)子區(qū)域。接著對(duì)各個(gè)區(qū)域預(yù)測多個(gè)錨點(diǎn)框(Bounding Box),并求出每個(gè)邊框的置信度(Confidence)。置信度分?jǐn)?shù)的大小,反映了錨點(diǎn)框內(nèi)存在目標(biāo)物體的準(zhǔn)確程度。置信度定義如下式:式中,Pr(Object)表示錨點(diǎn)框中存在目標(biāo)圖像的概率大小。IoU表示預(yù)測的目標(biāo)框與真實(shí)的標(biāo)注框之間的交并比,IoU的計(jì)算公式如下:
【參考文獻(xiàn)】:
期刊論文
[1]艦船目標(biāo)識(shí)別技術(shù)研究進(jìn)展[J]. 馬嘯,邵利民,金鑫,徐冠雷. 科技導(dǎo)報(bào). 2019(24)
[2]無人機(jī)海監(jiān)測繪技術(shù)應(yīng)用下艦船遙感圖像目標(biāo)檢測[J]. 安潔玉,丁斌芬. 艦船科學(xué)技術(shù). 2019(24)
[3]改進(jìn)的YOLO模型及其在艦船目標(biāo)識(shí)別中的應(yīng)用[J]. 馬嘯,邵利民,金鑫,徐冠雷. 電訊技術(shù). 2019(08)
[4]基于YOLO和嵌入式系統(tǒng)的車流量檢測[J]. 馬永杰,宋曉鳳. 液晶與顯示. 2019(06)
[5]基于Faster-RCNN和多分辨率SAR的海上艦船目標(biāo)檢測[J]. 胡炎,單子力,高峰. 無線電工程. 2018(02)
本文編號(hào):3601700
本文鏈接:http://sikaile.net/kejilunwen/chuanbolw/3601700.html
最近更新
教材專著