今天小李在逛人類知識寶庫時發現一個非常有意思的項目,“CoupleGenerator”, 可以理解爲伴侶生成器。哦也不對,單身的小夥伴不要想太多,國家不會分配,AI自然目前也是沒法大變活人,給你分配對象的。
那這個項目是做什麼的呢,準確的說是生成伴侶的樣貌圖片,通過一定夫妻合照的數據集進行訓練,你只要輸入你的照片便可以通過生成對抗網絡訓練出一個和你最爲般配的你的“愛人”的圖片。下圖是模型訓練了8800個步驟之後的結果,
環境配置
- tensorflow 1.1
- vgg 的預訓練權重 (爲了方便大家,我已經訓練好一個,文末可下載)
快速開始
- 下載訓練圖片並且解壓縮:https://cloudstor.aarnet.edu.au/plus/s/VWZJaWfbla3kFch (這裏要注意的是原作者把訓練圖片反向命名了,命名成了test)
- 克隆作者的Github倉庫
git clone https://github.com/irfanICMLL/CoupleGenerator
- 運行 autotest.sh 腳本
“這是兩年前的舊項目,她非常驚訝那麼多人會喜歡這個玩具項目”,作者在README上說道,“數據源來自網絡爬蟲所獲取的已婚夫婦的照片,並對想進行預處理,使用了 pix2pix 的代碼進行模型的訓練。”
這個爬蟲是一個百度圖片的小爬蟲。看得出來,21世紀的今天最重要的以及必須的就是數據,我們需要掌握一定的數據獲取的能力。我們知道,深度學習是由數據驅動的,也非常依賴於數據集,如果能收集到更多的訓練圖片,伴侶生成的效果會變得更加出色。
那麼大家還在等什麼?沒有對象的去GAN一個Couple,說不定人海茫茫你就能遇見那個他;有對象的去GAN一個小三,看好了什麼長相最有可能插足你們之間的感情!
哦對了,這裏是已經訓練好的vgg16模型。鏈接:https://pan.baidu.com/s/1NuKdxxsCdAKMMMLy6Yw5AA提取碼:ocnl