輕量化分割模型-LEDNet

????????LEDNet發(fā)表于2019年5月,LEDNet采用的是非對稱結(jié)構(gòu)的網(wǎng)絡(luò)來實(shí)現(xiàn)實(shí)時的語義分割。在編碼階段痘昌,編碼器采用了resnet作為backbone,每個resnet block利用通道分離和shuffle操作來降低計(jì)算量的同時保證了準(zhǔn)確率,另一方面辆苔,在解碼階段算灸,采用注意力金字塔網(wǎng)絡(luò)(APN)來進(jìn)一步的降低整個網(wǎng)絡(luò)的復(fù)雜性。

? ??????編碼器的核心單元是一個新穎的殘差模塊(如圖1.d所示驻啤。模塊取名SS-bnt)菲驴,該模塊利用跳躍連接以及通道split和shuffle的卷積。跳躍連接允許卷積學(xué)習(xí)有助于訓(xùn)練的殘差函數(shù)骑冗,split and shuffle操作增強(qiáng)了通道間的信息交換赊瞬,同時保持了與一維分解卷積相似的計(jì)算成本。在解碼器中贼涩,作者設(shè)計(jì)了一個注意力金字塔網(wǎng)絡(luò)(APN)來提取密集特征巧涧,而不是使用復(fù)雜的卷積[20]來提取密集特征,其中注意力機(jī)制用于估計(jì)每個像素的語義標(biāo)簽遥倦。


圖1 不同的殘差模塊設(shè)計(jì)

????????作者指出谤绳,根據(jù)參考文獻(xiàn)21所指出,1x1卷積網(wǎng)絡(luò)占據(jù)了大多數(shù)的計(jì)算資源袒哥,因此缩筛,在設(shè)計(jì)SS-bnt模塊時,作者取消了1x1卷積堡称,采用的是一維分解卷積(1x3瞎抛,3x1)來代替。

? ? ? ? 為了降低計(jì)算復(fù)雜度粮呢,在每個SS-nbt開始的時候婿失,輸入被分離到兩個低緯度的分支钞艇,每一個分支都獲得了輸入的一半通道數(shù)啄寡,為了避免1x1卷積,作者采用了專門的一維卷積(例如1x3哩照,3x1)挺物,最后兩個分支的卷積通過concatenate合并到一起,這樣飘弧,輸出和輸入的通道數(shù)目就保證了一致识藤。為了促進(jìn)訓(xùn)練,輸出還與上輸入進(jìn)行了相加次伶,然后痴昧,利用了文獻(xiàn)[21]的channel shuffle 操作,將輸出進(jìn)行channel shuffle 操作來保證兩個分支之間的信息交流冠王。這就是整個的SS-nbt模塊赶撰,一個SS-nbt完成后,輸出又作為下一個SS-nbt模塊的輸入。該模塊不僅高效豪娜,而且保證了準(zhǔn)確率餐胀。因?yàn)?.首先,每個SS-nbt的高效率使我們能夠使用更多的特征通道2.在每個SS-nbt單元中瘤载,將兩個分支組合起來的特征通道又進(jìn)行了隨機(jī)的shuffle操作否灾,又被送人到下一個SS-nbt模塊,這可以被事為是一種特征的重用鸣奔,在某種程度上擴(kuò)大了網(wǎng)絡(luò)的表現(xiàn)能力墨技,同時有沒有顯著的增加復(fù)雜性。

網(wǎng)絡(luò)的模型圖如圖2所示挎狸。


圖2 LEDNet網(wǎng)絡(luò)結(jié)構(gòu)圖

網(wǎng)絡(luò)中各個特征圖大小以及空洞卷積率如圖3所示健提。


圖3?

作者針對于編碼器中同樣采取了下采樣模塊,下采樣模塊是2個3x3卷積外加一個Maxpooling層構(gòu)成伟叛。

對于其解碼模塊私痹,文中的解釋不是很直觀,大家可以結(jié)合網(wǎng)上的復(fù)現(xiàn)源碼進(jìn)行理解统刮,就非常清楚了紊遵。在這里,我結(jié)合源碼給大家闡述一下解碼器部分的組成侥蒙。


圖4解碼部分詳解

如圖4所示暗膜,首先3,5鞭衩,7分別是不同的卷積和大小学搜,然后獲取得到不同感受野尺寸,之后论衍,在每個特征圖下接一個卷積網(wǎng)絡(luò)瑞佩,輸出通道數(shù)是類別數(shù)目,如圖4中的黑色模塊坯台。接著紫色的模塊是上采樣炬丸,然后“+”是特征圖之間逐元素的相加,綠色的特征圖模塊是通過逐元素相乘得到的蜒蕾,最右邊采用全局池化來獲得全局信息稠炬。


最后來展示一下LEDNet和其他網(wǎng)絡(luò)的效果對比。


圖5 不同網(wǎng)絡(luò)對比

圖5可以看出咪啡,LEDNet在精度上達(dá)到了最高的水準(zhǔn)首启,同時在推理速度和網(wǎng)絡(luò)模型大小方面也取得了不錯的成績。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末撤摸,一起剝皮案震驚了整個濱河市毅桃,隨后出現(xiàn)的幾起案子栽惶,更是在濱河造成了極大的恐慌,老刑警劉巖疾嗅,帶你破解...
    沈念sama閱讀 219,039評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件外厂,死亡現(xiàn)場離奇詭異,居然都是意外死亡代承,警方通過查閱死者的電腦和手機(jī)汁蝶,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,426評論 3 395
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來论悴,“玉大人掖棉,你說我怎么就攤上這事“蚬溃” “怎么了幔亥?”我有些...
    開封第一講書人閱讀 165,417評論 0 356
  • 文/不壞的土叔 我叫張陵,是天一觀的道長察纯。 經(jīng)常有香客問我帕棉,道長,這世上最難降的妖魔是什么饼记? 我笑而不...
    開封第一講書人閱讀 58,868評論 1 295
  • 正文 為了忘掉前任香伴,我火速辦了婚禮,結(jié)果婚禮上具则,老公的妹妹穿的比我還像新娘即纲。我一直安慰自己,他們只是感情好博肋,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,892評論 6 392
  • 文/花漫 我一把揭開白布低斋。 她就那樣靜靜地躺著,像睡著了一般匪凡。 火紅的嫁衣襯著肌膚如雪膊畴。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,692評論 1 305
  • 那天锹雏,我揣著相機(jī)與錄音巴比,去河邊找鬼术奖。 笑死礁遵,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的采记。 我是一名探鬼主播佣耐,決...
    沈念sama閱讀 40,416評論 3 419
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼唧龄!你這毒婦竟也來了兼砖?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,326評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎讽挟,沒想到半個月后懒叛,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,782評論 1 316
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡耽梅,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,957評論 3 337
  • 正文 我和宋清朗相戀三年薛窥,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片眼姐。...
    茶點(diǎn)故事閱讀 40,102評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡诅迷,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出众旗,到底是詐尸還是另有隱情罢杉,我是刑警寧澤,帶...
    沈念sama閱讀 35,790評論 5 346
  • 正文 年R本政府宣布贡歧,位于F島的核電站滩租,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏利朵。R本人自食惡果不足惜持际,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,442評論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望哗咆。 院中可真熱鬧蜘欲,春花似錦、人聲如沸晌柬。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,996評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽年碘。三九已至澈歉,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間屿衅,已是汗流浹背埃难。 一陣腳步聲響...
    開封第一講書人閱讀 33,113評論 1 272
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留涤久,地道東北人涡尘。 一個月前我還...
    沈念sama閱讀 48,332評論 3 373
  • 正文 我出身青樓,卻偏偏與公主長得像响迂,于是被迫代替她去往敵國和親考抄。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,044評論 2 355

推薦閱讀更多精彩內(nèi)容