1. 程式人生 > >風格遷移(2)-Fast Style Transfer

風格遷移(2)-Fast Style Transfer

info 圖片 參考 更新 ron bsp 深度 pytorch 們的

技術分享圖片

X為輸入圖片

fw 風格遷移的網絡

yc就是X

ys是風格後的圖片

y帽為輸入圖片X經過fw 風格遷移的網絡生成的圖片 y帽在內容上與yc相類似,在風格上與ys相類似。

Fast Style Transfer的訓練步驟如下:

1 輸入一張圖片x到fw中得到結果y帽

2 將y帽與yc輸入到loss network(VGG-16)中,計算它的relu3_3的輸出,並計算它們的均方誤差作為content loss

3 將y帽與ys輸入到loss network(VGG-16)中,計算它的relu1_2,relu2_2,relu3_3,relu4_3的輸出,再計算它們的Gram Matrix的均方誤差作為style loss

4 兩個損失相加,並反向傳播。更新fw的參數,固定loss network不動。

5 回到第一步,重新訓練fw.

簡單的數學原理如下:

技術分享圖片

技術分享圖片

技術分享圖片

源碼地址:https://github.com/jcjohnson/neural-style

參考博客及書籍:

1 深度學習框架pytorch入門及實踐 陳雲

2 https://blog.csdn.net/Hungryof/article/details/53981959

風格遷移(2)-Fast Style Transfer