論文學習16“Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network”文章學習

本文是CVPR17年的文章,也是超分辨領域的新作SRGAN,作者來自推特工作室,先放上整個網絡結構圖,再詳細介紹。

上周我在用SRCNN進行實驗效果不理想時,師兄推薦了這篇文章,事實上這篇文章和我上周看的SAGAN的整體結構很相似,都是在GAN的基礎上進行的image2image的工作。

在SRGAN中,作者論證PSNR不能作為評價超分辨的標準,因為這個和人對圖片的感受結果不一致,這里作者采用了MOS進行評價;另外作者在loss函數上進行了改進,更好的恢復出細節(jié)的紋理信息。接下來,就按照文章的順序進行介紹。


方法:

ISR是高分辨率圖像,ILR是低分辨率圖像,是由高分辨率圖像先加高斯噪聲然后經過一個r步長的下采樣得到的,所以高低分辨率的圖像大小分別是:rW*rH*C和W*H*C。模型的目的就是生成網絡G可以將輸入的低分辨率圖像重構出ISR,所以在生成器時就采用前饋CNN記作Gθ,參數是θ,那么此部分要優(yōu)化的就是:

和標準GAN一樣,模型需要訓練下式:


生成網絡的結構如下,核心是下面的B殘差塊部分,重復的殘差塊用于生成高分辨率圖像,然后接著兩個亞像素卷積層用于恢復高分辨率圖像的尺寸。

判別網絡的結構如下圖,連續(xù)的卷積層、BN和Leaky ReLU,緊接著是稠密塊和SIGMOD用于對圖像分類。

接著就是SRGAN的一個創(chuàng)新點,loss函數兩部分組成,內容損失和對抗損失:

首先內容損失,傳統(tǒng)的loss都是以像素為單位的MSE,如下式:

而MSE的loss函數使得輸出缺乏高頻成分,過于光滑不適宜人們閱讀,所以本文在基于預訓練的VGG19的RELU激活層來定義loss函數:

上式是通過歐氏距離定義的,其中

是VGG19里面第i個最大池層之后的第j個卷積層所提取到的特征,

是各自的特征維度。

對抗損失是指GAN網絡生成器部分是損失,

是生成圖像被判別為高分辨率圖像的概率:

實驗:

數據集是SET5、SET14和BSD100三個圖像超分辨領域的基準數據集,低分辨率和高分辨率之間是4倍的差距,采用PSNR和SSIM作為性能衡量標準。訓練的時候是用的ImageNet數據集下采樣,用adam優(yōu)化。

作者在衡量超分辨性能時提出了一個新的標準MOS,就是采用人作為標準,作者要求26個用戶對超分辨的圖像進行打分,1-5分,越高性能越好。在三個基準數據集的圖像進行打分,結果如下:

各個超分辨方法所得到的PSNR、SSIM和MOS的最終結果如下表,本文的生成網絡部分SRResNet得到了最高的指標,雖然SRGAN的PANR和SSIM并不是最高,但是他得到了最高的MOS值,這意味著PSNR和SSIM不能單獨作為超分辨的衡量標準。

作者針對內容損失函數也做了幾組對比實驗,分別在MSE、VGG22和VGG54的loss函數進行實驗,結果如下,雖然MSE的PSNR值更高,但是他在圖像的視覺感知上效果并不好,文理細節(jié)處理不夠精細,這又一次證明了作者對PSNR的觀點。

通過以上實驗看出,用均方誤差優(yōu)化SRResNet,能夠得到具有很高的峰值信噪比的結果。在訓練好的VGG模型的高層特征上計算感知損失來優(yōu)化SRGAN,并結合SRGAN的判別網絡,能夠得到峰值信噪比雖然不是最高,但是具有逼真視覺效果的結果,基于VGG模型高層特征比基于VGG模型低層特征的內容損失能生成更好的紋理細節(jié)。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 229,362評論 6 537
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現死者居然都...
    沈念sama閱讀 99,013評論 3 423
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 177,346評論 0 382
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,421評論 1 316
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,146評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,534評論 1 325
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,585評論 3 444
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,767評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體,經...
    沈念sama閱讀 49,318評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,074評論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,258評論 1 371
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,828評論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,486評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,916評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,156評論 1 290
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,993評論 3 395
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,234評論 2 375

推薦閱讀更多精彩內容