黄图鉴别工具 nsfw

Apache
Python
跨平台
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% 。

如有建议可邮件: zhengwu@midday.me

的码云指数为
超过 的项目
加载中

评论(1)

时光之舞丶
时光之舞丶
赞一个,可用

黄图鉴别工具 nsfw 首发

写在前面:不要找我要训练数据,我是遵纪守法的好公民,训练数据已经删除。 几天前看到 GitHub 上用于黄图鉴别的数据, 就 down 回来做了个黄图鉴别的模型,现模型已开源:nsfw 。 在线demo ...

01/23 10:28

没有更多内容

加载失败,请刷新页面

没有更多内容

1
回答
【一周热点】鉴黄训练数据已删除,你以为人民警察会相信你吗?

回顾一周社区热门资讯与你的精彩留言~ 第【十二】期:0119-0125 一周一问,你有什么见解? 有没有一本书或教程,教你如何把一堆乱代码逐步优化成逻辑清...

01/25 18:43

没有更多内容

加载失败,请刷新页面

没有更多内容

用Alamofire进行网络请求的一段代码解析(一)

/* 向服务器发送请求: request实际上有四个参数,但是后两个参数都设定为了默认值,所以在调用的时候只要初始化前两个参数就可以。 第一个参数是设定向服务器发起的请求的方法,通常为GET方...

2015/02/25 20:45
4.7K
0
web 题目

1. ssh 或 ssm 架构中,若自定义 filter 中 需要 调用 spring 容器中的对象,可以在filter 中 直接 使用 spring 注解 将 属性注入进来吗?为什么? 答: (1): 不可以,因为 filter 是在web...

2018/09/05 17:36
13
0
Log4j的错误处理

log4j

2016/04/12 09:29
9
1
Palace 120mm assault rifle in

Palace 120mm assault rifle in

2015/05/04 11:33
2
0
Coding and Paper Letter(三十二)

今年您的假期余额已不足,请好好学习。资源整理。老规矩第一篇Coding。 Coding: 1.一种超快速且可扩展的文件监视器,可在Linux,OSX和Windows上提供一致的界面。 nsfw 2.“R + OSM简介”研讨...

2018/10/09 21:35
9
0
【AI in 美团】如何基于深度学习实现图像的智能审核?

背景 美团每天有百万级的图片产生量,运营人员负责相关图片的内容审核,对涉及法律风险及不符合平台规定的图片进行删除操作。由于图片数量巨大,人工审核耗时耗力且审核能力有限。另外对于不...

2018/07/17 10:37
29
0
新手初识webservice-实践篇

理论背景:《新手初识webservice-理论篇》http://my.oschina.net/achi/blog/52766 编程语言:Java 必要技能:JavaWeb,MVC,Spring IOC 情景假设:你刚用Java做了个网上店铺,用了servlet,MVC...

2012/04/12 09:43
8.8K
1

没有更多内容

加载失败,请刷新页面

没有更多内容

返回顶部
顶部