栗子 发自 凹非寺
量子位 报道 | 公众号 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
— 完—
大咖齐聚!量子位MEET大会精彩回放
量子位 MEET 2020 智能未来大会精彩回放来袭!李开复、倪光南、景鲲、周伯文、吴明辉、曹旭东、叶杰平、唐文斌、王砚峰、黄刚、马原等AI大咖与你一起读懂人工智能。扫码观看回放吧~ ~
跟大咖交流 | 进入AI社群
量子位 QbitAI · 头条号签约作者
վ'ᴗ' ի 追踪AI技术和产品新动态
喜欢就点「在看」吧 !返回搜狐,查看更多
责任编辑: