1. 程式人生 > >基於Tensorflow的CycleGAN測試(非成對影象風格遷移:橙子--> 蘋果)

基於Tensorflow的CycleGAN測試(非成對影象風格遷移:橙子--> 蘋果)

影象風格遷移有兩種大的型別,一種是成對的,一種是非成對了。

成對的著名模型就是pix2pix,這種的例子,如從影像地圖轉換為向量地圖,從素描轉換為紋理圖等。這些的特點就是訓練資料集之間的成對的。

而非成對的,就是如從不同物體之間的轉換,如從橙子轉換為蘋果,或者不同季節之間的切換。

(http://img.mp.itc.cn/upload/20170527/6936b81df19f4038a076d41285e5d2f2_th.jpg,兩者之間的比較如這個網址的圖片所示。)


在上次實現了pix2pix之後,這回也嘗試著實現一下不成對的風格遷移,套路還是差不多的。著名就是cyclegan模型,這種模型也是需要從訓練集中(不同類的物體之間)尋找一種特徵對映,而這種對映的實現是基於GAN來實現的。

測試例子如下所示:


可以看到,在物體(巨集觀)之間的轉換,其結果是比較好的。

而一旦涉及到果實裡面的部分的比較時,這個就相對弱一些了。(效果不算好)