蕭箫 發自 凹非寺
量子位 | 公衆号 QbitAI
還在為找工作去理發店剃秃頭嗎?試試這個生成器!
隻要輸入一張照片,輸出的就是完美無瑕的光頭,氣質立刻從實習提升到主管(手動狗頭)
無論你是年輕的程式員小哥,還是長發飄飄的程式媛,都可以通過這個方法“變得更強”:
有了這個名叫HairMapper的“光頭神器”,P頭發就是分分鐘的事情,五官卻完全不受影響:
嗯,以後有了想做的發型,直接P一個……
要實作這麼好的效果,得用三個GAN合力才行。
所有人都能用的“光頭神器”
具體來說,HairMapper的原理一共分為三步,生成秃頭保留五官合并頭像。
首先,作者們利用StyleGAN,做出一個與原頭型相近的秃頭效果:
這時候我們看到,StyleGAN沒辦法做到精準地控制五官生成,是以臉型稍微有所變化。
那麼接下來,就是利用InterFaceGAN,摳出一個頭發以外的臉型和五官形象,同時也保留四周的風景:
最後,将前兩步生成的效果合成,就做出了一個完美的“光頭”:
不過值得一提的是,由于長發、女性光頭資料集少等原因,是以作者們額外為女版“光頭神器”多加了一個步驟。
在這其中,女性照片要先通過StyleFlow轉換成男性照片,再重複上述步驟,最終同樣得到一個光頭:
這樣的效果還是不錯的,在資料集中測試顯示,即使是發量驚人的女生也能一鍵脫發(doge)
作者們将HairMapper與其他的秃頭生成器做了個對比。
還做成了6000+秃頭資料集
事實上,之前并不是沒有人做過秃頭生成器,隻是效果并不太理想。
例如Adobe之前推出過的這個叫StyleFlow的生成器,就能讓馬斯克變秃:
但一個是頭發沒脫幹淨,另一個是五官也給GAN成了“另外一個人”。除了StyleFlow以外,其他類型的“秃頭生成器”也面臨這樣的問題:
這次的HairMapper在GAN脫發的基礎上,保留了原本的五官,使得還原出來的人臉非常真實。
無論男女老少或圓扁尖頭:
還是各種不同膚色乃至不同膚質:
甚至是側臉乃至有陰影,都能在HairMapper的作用下get絲滑的光頭特效:
目前作者們已經利用這個算法,将Flickr-Faces-HQ(FFHQ)資料集中的6000張人像變成了光頭。
以後無論是P頭發、還是變身男裝/女裝大佬,都能用這個開源資料集來訓練了:
作者介紹
一作Yiqian Wu,來自浙江大學計算機輔助設計與圖形學(CAD&CG)實驗室,目前在讀博士生,興趣是計算機視覺和人像編輯等。
二作Yong-Liang Yang,巴斯大學助理教授,本碩博畢業于清華大學,曾師從胡事民教授,研究方向是計算機圖形學、幾何模組化、VR/AR等。
通訊作者金小剛,浙大CAD&CG實驗室教授,研究方向包括電影特效模拟、自動駕駛仿真、創造性模組化等。
是不是迫不及待想要一試“光頭神器”了?
據作者表示,目前HairMapper的預訓練代碼還在加急制作中,馬上就會放出來,而資料集已經可以用了~