風格遷移(2)-Fast Style Transfer
阿新 • • 發佈:2018-06-01
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