黄图鉴别工具 nsfw 首发

来源:投稿
作者: 正_午
2019年01月23日

写在前面:不要找我要训练数据,我是遵纪守法的好公民,训练数据已经删除。

几天前看到 GitHub 上用于黄图鉴别的数据, 就 down 回来做了个黄图鉴别的模型,现模型已开源:nsfw在线demo

数据

训练数据共五个类别:

porn - pornography images hentai - hentai images, but also includes pornographic drawings sexy - sexually explicit images, but not pornography. Think nude photos, playboy, bikini, beach volleyball, etc. neutral - safe for work neutral images of everyday things and people drawings - safe for work drawings (including anime)

训练数据的数量如下:

drawing: 18251 hentai: 42649 netural: 44677 porn: 108114 sexy: 17253

使用

可以直接下载代码和训练好的模型来用 下载代码和模型

git clone https://github.com/rockyzhengwu/nsfw.git

训练好的模型在 /data/目录下。

cd nsfw
python nsfw_predict.py /tmp/test/test.jpeg

输出结果:

{'class': 'sexy', 'probability': {'drawings': 0.008320281, 'hentai': 0.0011919827, 'neutral': 0.13077603, 'porn': 0.13146976, 'sexy': 0.72824186}}

class: 图片所属列表 probability: 各类别所属的概率得分

也可以使用 Tensorflow Serving 部署,详情可以查看nsfw

模型

模型部分采用 Resnet ,主要参考了 Tensorflow 官方的代码 做了少许修改。详情见 resnet目录下。 尝试了微调和训练一个相对简单的新模型,微调的效果并不好,反而速度会慢很多。我想是因为这些图片和ImageNet 数据集相差太大的缘故。最终发布的模型使用Resnet-32, 准确率 90% 。

展开阅读全文
134 收藏
分享
加载中
精彩评论
训练数据已删除,你以为我们人民警察会相信你吗😂
2019-01-23 10:38
25
举报
我想建立一个开源项目,专门提供这个的训练数据,请问应该托管到那家?
2019-01-23 12:18
24
举报
https://github.com/alexkimxyz/nsfw_data_scrapper/tree/master/raw_data github已经成了老司机的飙车场地
2019-01-23 11:07
18
举报
高举社会主义精神文明大旗!向高技术化社会主义精神文明致敬!向伟大的精神乌托邦前进!
2019-01-23 10:44
11
举报
https://github.com/alexkimxyz/nsfw_data_scrapper/blob/master/raw_data/sexy/urls_sexy.txt害人不浅
2019-01-23 17:32
7
举报
最新评论 (40)
贼喊捉贼,已举报110
2019-01-28 10:03
0
回复
举报
急需训练集
2019-01-28 09:56
0
回复
举报
niaoshe666你想要的哦
2019-01-26 08:13
0
回复
举报
你懂的味,niaoshe666
2019-01-26 08:11
0
回复
举报
老乡开门,查水表
2019-01-25 17:51
0
回复
举报

引用来自“郭大鹏”的评论

我想建立一个开源项目,专门提供这个的训练数据,请问应该托管到那家?
秀儿
2019-01-24 10:28
0
回复
举报

引用来自“开源中国首席一失足成千古风流人物以稀为贵”的评论

谁有图片资源...😃
种子更可贵
2019-01-23 22:59
0
回复
举报
有没有视频鉴别工具的数据集😆
2019-01-23 22:07
0
回复
举报
_-
谁有图片资源...😃
2019-01-23 19:08
1
回复
举报
This file is too big to show. Sorry!
2019-01-23 17:34
0
回复
举报
更多评论
40 评论
134 收藏
分享
OSCHINA
登录后可查看更多优质内容
返回顶部
顶部