圖像超分辨率重建之SRCNN

筆者從CSDN平臺(tái)遷移至簡書:https://blog.csdn.net/Autism_/article/details/79401798(CSDN鏈接)

圖像超分辨率重建:指通過低分辨率圖像或圖像序列恢復(fù)出高分辨率圖像苗缩。高分辨率圖像意味著圖像具有更多的細(xì)節(jié)信息、更細(xì)膩的畫質(zhì),傅联,這些細(xì)節(jié)在高清電視跌帐、醫(yī)學(xué)成像合溺、遙感衛(wèi)星成像等領(lǐng)域有著重要的應(yīng)用價(jià)值。

Super-Resolution Convolutional Neural Network:本篇文章講述的是深度學(xué)習(xí)在圖像超分辨率重建問題的開山之作SRCNN(Super-Resolution Convolutional Neural Network)。香港中文大學(xué)Dong等將卷積神經(jīng)網(wǎng)絡(luò)應(yīng)用于單張圖像超分辨率重建上(Image Super-Resolution Using Deep Convolutional Networks)

論文與官方代碼: paper and offical code

筆者復(fù)現(xiàn)代碼(Tensorflow version):my code


一逢勾、SRCNN

1. SRCNN框架

圖1 SRCNN框架

圖1為SRCNN的框架扰柠,SRCNN將深度學(xué)習(xí)與傳統(tǒng)稀疏編碼之間的關(guān)系作為依據(jù)粉铐,將3層網(wǎng)絡(luò)劃分為圖像塊提取(Patch extraction and representation)、非線性映射(Non-linear mapping)以及最終的重建(Reconstruction)卤档。

2. SRCNN流程

? ? (1)先將低分辨率圖像使用雙三次差值放大至目標(biāo)尺寸(如放大至2倍蝙泼、3倍、4倍)劝枣,此時(shí)仍然稱放大至目標(biāo)尺寸后的圖像為低分辨率圖像(Low-resolution image)汤踏,即圖中的輸入(input)织鲸;

? ? (2)將低分辨率圖像輸入三層卷積神經(jīng)網(wǎng)絡(luò),(舉例:在論文中的其中一實(shí)驗(yàn)相關(guān)設(shè)置溪胶,對YCrCb顏色空間中的Y通道進(jìn)行重建搂擦,網(wǎng)絡(luò)形式為(conv1+relu1)—(conv2+relu2)—(conv3))第一層卷積:卷積核尺寸9×9(f1×f1),卷積核數(shù)目64(n1)哗脖,輸出64張?zhí)卣鲌D瀑踢;第二層卷積:卷積核尺寸1×1(f2×f2),卷積核數(shù)目32(n2)才避,輸出32張?zhí)卣鲌D橱夭;第三層卷積:卷積核尺寸5×5(f3×f3),卷積核數(shù)目1(n3)工扎,輸出1張?zhí)卣鲌D即為最終重建高分辨率圖像徘钥。

3. 如何訓(xùn)練?

(1)訓(xùn)練數(shù)據(jù)集:論文中某一實(shí)驗(yàn)采用91張自然圖像作為訓(xùn)練數(shù)據(jù)集肢娘,對訓(xùn)練集中的圖像先使用雙三次差值縮小到低分辨率尺寸呈础,再將其放大到目標(biāo)放大尺寸,最后切割成諸多33×33圖像塊作為訓(xùn)練數(shù)據(jù)橱健,作為標(biāo)簽數(shù)據(jù)的則為圖像中心的21×21圖像塊(與卷積層細(xì)節(jié)設(shè)置相關(guān))而钞;

(2)損失函數(shù):采用MSE函數(shù)作為卷積神經(jīng)網(wǎng)絡(luò)損失函數(shù);

(3)卷積層細(xì)節(jié)設(shè)置:第一層卷積核9×9拘荡,得到特征圖尺寸為(33-9)/1+1=25臼节,第二層卷積核1×1,得到特征圖尺寸不變珊皿,第三層卷積核5×5网缝,得到特征圖尺寸為(25-5)/1+1=21。訓(xùn)練時(shí)得到的尺寸為21×21蟋定,因此圖像中心的21×21圖像塊作為標(biāo)簽數(shù)據(jù)粉臊。(卷積訓(xùn)練時(shí)不進(jìn)行padding)

4. 如何測試?

(1)全卷積網(wǎng)絡(luò):所用網(wǎng)絡(luò)為全卷積網(wǎng)絡(luò)驶兜,因此作為實(shí)際測試時(shí)扼仲,直接輸入完整圖像即可;

(2)Padding:訓(xùn)練時(shí)得到的實(shí)際上是除去四周(33-21)/2=6像素外的圖像抄淑,若直接采用訓(xùn)練時(shí)的設(shè)置(無padding)屠凶,得到的圖像最后會(huì)減少四周各6像素(如插值放大后輸入512×512,輸出500×500)肆资。因此在測試時(shí)每一層卷積都進(jìn)行了padding(卷積核尺寸為1×1的不需要進(jìn)行padding)矗愧。這樣保證插值放大后輸入與輸出尺寸的一致性。

? ? (使用Tensorflow進(jìn)行復(fù)現(xiàn)時(shí)迅耘,圖像預(yù)處理時(shí)將像素點(diǎn)取值歸一化至[0,1]贱枣,測試時(shí)监署,得到的最后一層特征圖即重建結(jié)果直接乘以255再使用uint8轉(zhuǎn)換時(shí)為0-255取值時(shí)會(huì)出現(xiàn)一些問題颤专,如左下圖2中方框所示纽哥,因此在乘以255前,將負(fù)值設(shè)置為0栖秕,大于255的設(shè)置為255春塌,再使用uint轉(zhuǎn)換即可解決


圖2? 不對負(fù)值及大于1的值進(jìn)行處理的結(jié)果圖
圖3? 對負(fù)值及大于1的值進(jìn)行處理的結(jié)果圖

(詳細(xì)的padding方式等,可以查看鏈接中的代碼)

5. 重建結(jié)果簇捍?

(1)客觀評價(jià)指標(biāo)PSNR與SSIM:相比其他傳統(tǒng)方法只壳,SRCNN取得更好的重建效果

表1 客觀效果

(2)主觀效果:相比其他傳統(tǒng)方法,SRCNN重建效果更具優(yōu)勢

圖4 主觀效果
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末暑塑,一起剝皮案震驚了整個(gè)濱河市吼句,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌事格,老刑警劉巖惕艳,帶你破解...
    沈念sama閱讀 221,548評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異驹愚,居然都是意外死亡远搪,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,497評論 3 399
  • 文/潘曉璐 我一進(jìn)店門逢捺,熙熙樓的掌柜王于貴愁眉苦臉地迎上來谁鳍,“玉大人,你說我怎么就攤上這事劫瞳√惹保” “怎么了?”我有些...
    開封第一講書人閱讀 167,990評論 0 360
  • 文/不壞的土叔 我叫張陵志于,是天一觀的道長涮因。 經(jīng)常有香客問我,道長恨憎,這世上最難降的妖魔是什么蕊退? 我笑而不...
    開封第一講書人閱讀 59,618評論 1 296
  • 正文 為了忘掉前任,我火速辦了婚禮憔恳,結(jié)果婚禮上瓤荔,老公的妹妹穿的比我還像新娘。我一直安慰自己钥组,他們只是感情好输硝,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,618評論 6 397
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著程梦,像睡著了一般点把。 火紅的嫁衣襯著肌膚如雪橘荠。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,246評論 1 308
  • 那天郎逃,我揣著相機(jī)與錄音哥童,去河邊找鬼。 笑死褒翰,一個(gè)胖子當(dāng)著我的面吹牛贮懈,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播优训,決...
    沈念sama閱讀 40,819評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼朵你,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了揣非?” 一聲冷哼從身側(cè)響起抡医,我...
    開封第一講書人閱讀 39,725評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎早敬,沒想到半個(gè)月后忌傻,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,268評論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡搁嗓,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,356評論 3 340
  • 正文 我和宋清朗相戀三年芯勘,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片腺逛。...
    茶點(diǎn)故事閱讀 40,488評論 1 352
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡荷愕,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出棍矛,到底是詐尸還是另有隱情安疗,我是刑警寧澤,帶...
    沈念sama閱讀 36,181評論 5 350
  • 正文 年R本政府宣布够委,位于F島的核電站荐类,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏茁帽。R本人自食惡果不足惜玉罐,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,862評論 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望潘拨。 院中可真熱鬧吊输,春花似錦、人聲如沸铁追。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,331評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至扭屁,卻和暖如春算谈,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背料滥。 一陣腳步聲響...
    開封第一講書人閱讀 33,445評論 1 272
  • 我被黑心中介騙來泰國打工然眼, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人幔欧。 一個(gè)月前我還...
    沈念sama閱讀 48,897評論 3 376
  • 正文 我出身青樓罪治,卻偏偏與公主長得像丽声,于是被迫代替她去往敵國和親礁蔗。 傳聞我的和親對象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,500評論 2 359

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