只要輸入一張照片，輸出的就是完美無瑕的光頭，氣質(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之家所有文章均包含本聲明。