浙大小姐姐搞出禿頭生成器登CVPR，完美保留五官臉型，這就給簡(jiǎn)歷加加分

蕭簫 發(fā)自 凹非寺

量子位 | 公眾號(hào) QbitAI

還在為找工作去理發(fā)店剃禿頭嗎？試試這個(gè)生成器！

只要輸入一張照片，輸出的就是完美無瑕的光頭，氣質(zhì)立刻從實(shí)習(xí)提升到主管（手動(dòng)狗頭）

無論你是年輕的程序員小哥，還是長(zhǎng)發(fā)飄飄的程序媛，都可以通過這個(gè)方法“變得更強(qiáng)”：

有了這個(gè)名叫HairMapper的“光頭神器”，P頭發(fā)就是分分鐘的事情，五官卻完全不受影響：

嗯，以后有了想做的發(fā)型，直接P一個(gè)……

要實(shí)現(xiàn)這么好的效果，得用三個(gè)GAN合力才行。

所有人都能用的“光頭神器”

具體來說，HairMapper的原理一共分為三步，生成禿頭保留五官合并頭像。

首先，作者們利用StyleGAN，做出一個(gè)與原頭型相近的禿頭效果：

這時(shí)候我們看到，StyleGAN沒辦法做到精準(zhǔn)地控制五官生成，因此臉型稍微有所變化。

那么接下來，就是利用InterFaceGAN，摳出一個(gè)頭發(fā)以外的臉型和五官形象，同時(shí)也保留四周的風(fēng)景：

最后，將前兩步生成的效果合成，就做出了一個(gè)完美的“光頭”：

不過值得一提的是，由于長(zhǎng)發(fā)、女性光頭數(shù)據(jù)集少等原因，因此作者們額外為女版“光頭神器”多加了一個(gè)步驟。

在這其中，女性照片要先通過StyleFlow轉(zhuǎn)換成男性照片，再重復(fù)上述步驟，最終同樣得到一個(gè)光頭：

這樣的效果還是不錯(cuò)的，在數(shù)據(jù)集中測(cè)試顯示，即使是發(fā)量驚人的女生也能一鍵脫發(fā)（doge）

作者們將HairMapper與其他的禿頭生成器做了個(gè)對(duì)比。

還做成了6000+禿頭數(shù)據(jù)集

事實(shí)上，之前并不是沒有人做過禿頭生成器，只是效果并不太理想。

例如Adobe之前推出過的這個(gè)叫StyleFlow的生成器，就能讓馬斯克變禿：

但一個(gè)是頭發(fā)沒脫干凈，另一個(gè)是五官也給GAN成了“另外一個(gè)人”。除了StyleFlow以外，其他類型的“禿頭生成器”也面臨這樣的問題：

這次的HairMapper在GAN脫發(fā)的基礎(chǔ)上，保留了原本的五官，使得還原出來的人臉非常真實(shí)。

無論男女老少或圓扁尖頭：

還是各種不同膚色乃至不同膚質(zhì)：

甚至是側(cè)臉乃至有陰影，都能在HairMapper的作用下get絲滑的光頭特效：

目前作者們已經(jīng)利用這個(gè)算法，將Flickr-Faces-HQ（FFHQ）數(shù)據(jù)集中的6000張人像變成了光頭。

以后無論是P頭發(fā)、還是變身男裝/女裝大佬，都能用這個(gè)開源數(shù)據(jù)集來訓(xùn)練了：

作者介紹

一作Yiqian Wu，來自浙江大學(xué)計(jì)算機(jī)輔助設(shè)計(jì)與圖形學(xué)（CAD&CG）實(shí)驗(yàn)室，目前在讀博士生，興趣是計(jì)算機(jī)視覺和人像編輯等。

二作Yong-Liang Yang，巴斯大學(xué)助理教授，本碩博畢業(yè)于清華大學(xué)，曾師從胡事民教授，研究方向是計(jì)算機(jī)圖形學(xué)、幾何建模、VR/AR等。

通訊作者金小剛，浙大CAD&CG實(shí)驗(yàn)室教授，研究方向包括電影特效模擬、自動(dòng)駕駛仿真、創(chuàng)造性建模等。

是不是迫不及待想要一試“光頭神器”了？

據(jù)作者表示，目前HairMapper的預(yù)訓(xùn)練代碼還在加急制作中，馬上就會(huì)放出來，而數(shù)據(jù)集已經(jīng)可以用了~

關(guān)鍵詞： 浙大小姐姐搞出禿頭生成器登CVPR 完美保留五官臉型