只要輸入一張照片,輸出的就是完美無瑕的光頭,氣質(zhì)立刻從實習提升到主管(手動狗頭)
無論你是年輕的程序員小哥,還是長發(fā)飄飄的程序媛,都可以通過這個方法“變得更強”:
有了這個名叫 HairMapper 的“光頭神器”,P 頭發(fā)就是分分鐘的事情,五官卻完全不受影響:
嗯,以后有了想做的發(fā)型,直接 P 一個……
要實現(xiàn)這么好的效果,得用三個 GAN 合力才行。
所有人都能用的“光頭神器”
具體來說,HairMapper 的原理一共分為三步,生成禿頭 → 保留五官 → 合并頭像。
首先,作者們利用 StyleGAN,做出一個與原頭型相近的禿頭效果:
這時候我們看到,StyleGAN 沒辦法做到精準地控制五官生成,因此臉型稍微有所變化。
那么接下來,就是利用 InterFaceGAN,摳出一個頭發(fā)以外的臉型和五官形象,同時也保留四周的風景:
最后,將前兩步生成的效果合成,就做出了一個完美的“光頭”:
不過值得一提的是,由于長發(fā)、女性光頭數(shù)據(jù)集少等原因,因此作者們額外為女版“光頭神器”多加了一個步驟。
在這其中,女性照片要先通過 StyleFlow 轉(zhuǎn)換成男性照片,再重復上述步驟,最終同樣得到一個光頭:
這樣的效果還是不錯的,在數(shù)據(jù)集中測試顯示,即使是發(fā)量驚人的女生也能一鍵脫發(fā)(doge)
作者們將 HairMapper 與其他的禿頭生成器做了個對比。
還做成了 6000 + 禿頭數(shù)據(jù)集
事實上,之前并不是沒有人做過禿頭生成器,只是效果并不太理想。
例如 Adobe 之前推出過的這個叫 StyleFlow 的生成器,就能讓馬斯克變禿:
但一個是頭發(fā)沒脫干凈,另一個是五官也給 GAN 成了“另外一個人”。除了 StyleFlow 以外,其他類型的“禿頭生成器”也面臨這樣的問題:
這次的 HairMapper 在 GAN 脫發(fā)的基礎上,保留了原本的五官,使得還原出來的人臉非常真實。
無論男女老少或圓扁尖頭:
還是各種不同膚色乃至不同膚質(zhì):
甚至是側(cè)臉乃至有陰影,都能在 HairMapper 的作用下 get 絲滑的光頭特效:
目前作者們已經(jīng)利用這個算法,將 Flickr-Faces-HQ(FFHQ)數(shù)據(jù)集中的 6000 張人像變成了光頭。
以后無論是 P 頭發(fā)、還是變身男裝 / 女裝大佬,都能用這個開源數(shù)據(jù)集來訓練了:
作者介紹
一作 Yiqian Wu,來自浙江大學計算機輔助設計與圖形學(CAD&CG)實驗室,目前在讀博士生,興趣是計算機視覺和人像編輯等。
二作 Yong-Liang Yang,巴斯大學助理教授,本碩博畢業(yè)于清華大學,曾師從胡事民教授,研究方向是計算機圖形學、幾何建模、VR / AR 等。
通訊作者金小剛,浙大 CAD&CG 實驗室教授,研究方向包括電影特效模擬、自動駕駛仿真、創(chuàng)造性建模等。
是不是迫不及待想要一試“光頭神器”了?
據(jù)作者表示,目前 HairMapper 的預訓練代碼還在加急制作中,馬上就會放出來,而數(shù)據(jù)集已經(jīng)可以用了~
項目地址:
https://github.com/oneThousand1000/non-hair-FFHQ
論文地址:
http://www.cad.zju.edu.cn/home/jin/cvpr2022/HairMapper.pdf
廣告聲明:文內(nèi)含有的對外跳轉(zhuǎn)鏈接(包括不限于超鏈接、二維碼、口令等形式),用于傳遞更多信息,節(jié)省甄選時間,結(jié)果僅供參考,IT之家所有文章均包含本聲明。