1. 程式人生 > >invertible Conditional GANs for image editing

invertible Conditional GANs for image editing

github程式碼為https://github.com/Guim3/IcGAN

通常GAN的生成網路輸入為一個噪聲向量z,文獻的創新點是,利用一個encoder網路,對輸入影象提取得到一個特徵向量z,將特徵向量z,以及需要轉換的目標attribute向量y串聯輸入生成網路,得到生成影象,網路結構如下,

這裡寫圖片描述

如上圖所示,包括連個encoder網路,分別為 Ez,Ey , Ez 用於將輸入影象編碼為特徵向量z,也就是特徵向量z提取了影象的基本特徵. Ey 用於將輸入影象編碼為特徵向量y,特徵向量y包括影象的性別,是否為black hair,brown hair,是否make-up,sungalsses等資訊,

y 為我們需要生成的目標影象的特徵向量,將特徵向量y的對於位置的特徵值轉換,得到目標特徵向量 y ,將z與 y 串聯,輸入生成網路,則可以生成含有目標特徵的影象.

encoder網路引數如下,

這裡寫圖片描述

之後將生成影象,目標影象分別輸入判別網路,生成網路,判別網路結構圖如下,

這裡寫圖片描述

生成,判別網路引數為,

這裡寫圖片描述

生成效果,

這裡寫圖片描述