AI識別二次元妹子「絕對領域」，雙馬尾、超短裙、長筒襪也能檢測 | 開源+線上玩

• 2020 年 1 月 21 日
• 筆記

栗子 發自 凹非寺 量子位 報道 | 公眾號 QbitAI

有一隻AI，名叫Deep Danbooru，專門用來審視二次元妹子身上的各種特徵。

當然，首要前提是看出一個人物是不是妹子。

比如，輸入一隻性別特徵不明的千戶：

△來自《少女終末旅行》

妹子判斷成功，置信度0.92。

再輸入一隻長頭髮的幸平老爹：

△ 來自《食戟之靈》

漢子判斷成功，置信度0.97。

兩道並不簡單的題目，都沒有難住AI。

接下來，到了正經觀察妹子的時間：

絕對領域

讓我們輸入一隻Miku吧，看AI都能發現什麼：

萬萬沒想到，除了以100%置信度判斷為妹子之外，AI還看出了25種其他特徵 (並認出了她是Miku) ：

1水藍色眼睛；2水藍色頭髮；3露肩；4腿着黑色；5黑色短裙；6袖套；8頭戴耳機；9長發；10看向觀眾；11指甲油；12領帶；13褶裙 14襯衫；15簡單背景；16無袖；17無袖襯衫；18圍笑；19單人；20過膝靴；21黑色長筒襪；22雙馬尾；23超長頭髮；24白色背景； 25絕對領域 (zettai ryouiki) ！！！

果然是只敬業的AI，每個細節都不放過。

就連褲襪和短裙之間那一抹珍貴的風景，也毫不吝嗇地識別出來，完全沒有保留。

△ 絕對領域

這樣一來，只要有個豐盛的數據集，不論你鍾愛哪一種特徵，大概都可以批量篩選出來，盡情觀賞了。

說到數據集

如果你還沒有這樣的資源，這隻AI的作者Kichang Kim推薦了一個站點，就叫「Danbooru」： https://danbooru.donmai.us/

如果你不知道這裡的妹子怎樣批量下載，作者還為你開發了一枚下載器： https://github.com/KichangKim/DeepDanbooru

下載好圖像之後，就要給圖像加標籤。作者自己用了6000個標籤，也是來自上面的站點，可以直接下載來用：

> python program.py download-tags [your_project_folder]

另外，假如你對某些標籤有偏好，也可以按照自己的需求過濾一下數據集：

> python program.py make-training-database [your_dataset_sqlite_path] [your_filtered_sqlite_path]

當然，也可以自己造數據集，作者在項目頁提供了數據結構。

數據集的問題解決了，然後就是：

訓練模型吧

DeepDanbooru是一個基於ResNet的算法。

開啟訓練，就用這行代碼：

> python program.py train-project [your_project_folder]

作者是用Ryzen 7 1800X + Geforce 1070，花了28天訓練成功的。

如果你不想訓練的話，作者表示也有預訓練的模型提供，可直接食用。

訓練完畢，就可以享用了：

> python program.py evaluate-project [your_project_folder] [image_file_path]

還能幹什麼

一旦AI對二次元妹子的特徵了如指掌，就可以用它來定製老婆。

算法的作者也已經開始這樣做了：

你看，滿屏的參數都可以調，還愁找不到自己喜歡的妹子？

所以，只要有想法就去行動吧，少年。

代碼傳送門： https://github.com/KichangKim/DeepDanbooru Demo傳送門： http://kanotype.iptime.org:8003/deepdanbooru/ 數據集下載器： https://github.com/KichangKim/DanbooruDownloader

— 完 —