动漫妹子分类

作者: kurumi233 | 来源:发表于2018-02-25 14:47 被阅读0次

动漫妹子分类
动漫妹子
临摹无名
动漫推荐：这部番全部由妹子构成，世界靠妹子来守护？
国产动漫《镇魂街》
动漫少女怎么画身体？
板绘练习--18--Darling in the frankxx
盘点动漫中的那些黑客妹子！
Python爬虫 - 抓取divnil动漫妹子图
动漫：这些妹子助攻别人苦了自己

这是我闲来无事做的一个小项目，根据输入的动漫女性图片，可以判断女性的类别，暂时只是大概做了三类：萝莉，御姐，成熟女性。如果你们感兴趣的话，可以添加更多的分类，比如黑长直，金发等，当然也可以是隐藏属性等等，比如天然呆，傲娇，腹黑，病娇等等，当然做到这些需要你好好的准备数据集。

数据集

我的数据集其实没有花太多时间，简单的用爬虫爬取百度图片，三种类型，其中前两种比较好爬，爬取完后进行筛选，去除背景较大的，尽量让人物填满整个图片，当然如果你有心的话，可以手动截取，去除多余的背景，我每一类只留下了50张图片，所以截取工作量也能接受。
当然你如果想获得更好的效果，就需要更多的数据。第三类图片爬取成熟女性的动漫角色的话，出来的基本都是很奇妙的东西。。我试了下爬取K站的图，也都是不太合适的图片，因此我不得不手动从百度图片找，大致让我满意的就找了30张。。也懒得再找了，就这样吧。（真的是懒）

PS：因为动漫角色不是人脸和真实物体，所以我暂时不太清楚有什么办法可以程序式的识别出图片中的角色，如果是人脸的话，有很多的程序方法可以直接将人脸从图片中扣出。所以稳妥点还是手动截取角色吧，但是我比较懒2333我直接用了爬虫爬取后的图片，然后简单筛选一下，留下50张，之后我会上传到百度云盘。所以我这个数据集不管在数量还是质量上都是很差的，你们有很大的提升空间

Add：数据集下载提取码：2nw9 考虑到墙的原因，就不适用GoogleDrive了。

数据增强

要训练神经网络，一共130张图片肯定是不够的，因此需要数据增强。我提供了一些利用PIL模块数据增强的方法，放在了process.py中，你们可以自行修改，也可以直接用。

最后我增强到了1040张图片，勉勉强强合格了。

训练

训练相关的配置都在Config中：

class Config(object):
    model_saved_path='checkpoints/anime_type.pkl'
    train_path='data/'
    img_size = 128
    batch_size=20
    shuffle = True
    num_workers = 2
    lr=1e-3
    use_gpu=False
    epoch = 20
    test_img = 'test/luo1.jpg'

你们可以根据实际情况进行修改。我这里只训练了20轮，又因为图片数量不大，所以只用了10分钟不到就训练好了，因此我就不给出checkpoints文件了，你们可以修改配置，增加epoch，制作更大更好的数据集，这样效果会更好。