软件简介

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

几天前看到 GitHub 上用于黄图鉴别的数据, 就 down 回来做了个黄图鉴别的模型,现模型已开源,在线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

展开阅读全文

代码

的 Gitee 指数为
超过 的项目

评论 (1)

加载中
赞一个,可用
2019/01/30 15:03
回复
举报
更多评论
发表了资讯
2019/01/23 10:29

黄图鉴别工具 nsfw 首发

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

40
134
没有更多内容
加载失败,请刷新页面
点击加载更多
加载中
下一页
发表于AI & 大数据专区
2018/02/01 13:50

[NSFW]deepfakes开源了,你也可以拍成人电影了

背景 这个主题 r/deepfakes 在reddit上已经爆红一个多月了,可以百度下 ‘AI算法把女星面部嫁接到成人电影上:效果以假乱真’这篇文章有详细的介绍。 昨天主题的管理者在github上开源了自己的代码,并在reddit上新...

0
1
发表于云计算专区
04/13 22:39

深夜,使用NSFW尝试一下图片鉴黄

前几天公司在讨论鉴黄的问题,对接了一些厂家提供的鉴黄服务。由于公司本身就是做音视频领域相关的,鉴黄的需求量较大。同时秉着「好好学习,天天向上」的精神。想自己试一下鉴黄相关的东西。刚好同事也提出了一个...

0
0
发表了博客
08/06 11:41

正道的光!这有个用TensorFlow做的小黄图过滤器

  机器之心报道   编辑:张倩、陈萍       有了这个插件,再也不怕同事误会了。   相信每个人都遇到过这种情况:想下载一部电影或一个软件,结果跳出一堆色情广告,因为怕同学或同事误会,不得不赶紧关掉浏览器。这种情况广泛存在于电子书、电影等资源类网站以及插件、模板等工具类网站,给广大网友带来了很大...

0
0
发表于软件架构专区
08/06 09:00

正道的光!这有个用TensorFlow做的小黄图过滤器

机器之心报道 编辑:张倩、陈萍 有了这个插件,再也不怕同事误会了。 相信每个人都遇到过这种情况:想下载一部电影或一个软件,结果跳出一堆色情广告,因为怕同学或同事误会,不得不赶紧关掉浏览器。这种情况广泛...

0
0
发表于AI & 大数据专区
02/12 22:20

158万张图像的鉴黄数据集,这份学习资料你扛得住吗?

本文经机器之心(almosthuman2014) 授权转载,禁止二次转载 如果你想训练一个内容审核系统过滤不合适的信息,或用 GAN 实现一些大胆的想法,那么数据集是必不可少的。但限制级图像很难收集,也很少会开源。在这个...

0
0
发表了博客
08/07 12:26

正道的光!这有个用TensorFlow做的小黄图过滤器 - 知乎

有了这个插件,再也不怕同事误会了。 机器之心报道,编辑:张倩、陈萍。 相信每个人都遇到过这种情况:想下载一部电影或一个软件,结果跳出一堆色情广告,因为怕同学或同事误会,不得不赶紧关掉浏览器。这种情况广泛存在于电子书、电影等资源类网站以及插件、模板等工具类网站,给广大网友带来了很大困扰。 因此,很多人...

0
0
发表了博客
2019/09/27 00:38

jquery+flask+keras+nsfw快速搭建一个简易鉴黄工具

1. demo 地址:http://www.huchengchun.com:8127/porn_classification 接口说明: 1. http://www.huchengchun.com:8127/porn_classification 提供了一个简易的网页工具,用户可以上传若干张图片,服务端会传回每一张图片是否是色情图片的判定,结果的形式是json格式。截图如下: 返回结果的说明: 返回结果整体是一个json,ke...

0
0
发表了博客
2019/04/10 10:10

机器学习 TensorFlow 实现智能鉴黄

前言 最近在做一款图床服务,关注公号的小伙伴一定记得小柒曾说过,会在周末放出的,不好意思放大家鸽子了。之所以一直没敢放出,是因为鉴黄接口一直没调试好,虽然我对公号的小伙伴百分之百信任,奈何互联网鱼龙混杂,万一上传了什么不雅的图片,然后不巧被某部门发现了,我包括我的服务器域名可就彻底玩完了! 架构 如...

0
0
发表于程序人生专区
2019/11/22 10:20

深夜吐血训练了100万小黄图撸了一个鉴黄图床

点击▲关注 “爪哇笔记” 给公众号标星置顶 更多精彩 第一时间直达 一、前言 最近在做一款图床服务,关注公号的小伙伴一定记得小柒曾说过,会在周末放出的,不好意思放大家鸽子了。之所以一直没敢放出,是因为鉴黄...

0
0
发表了博客
10/29 07:20

机器学习 TensorFlow 实现智能鉴黄

前言 最近在做一款图床服务,关注公号的小伙伴一定记得小柒曾说过,会在周末放出的,不好意思放大家鸽子了。之所以一直没敢放出,是因为鉴黄接口一直没调试好,虽然我对公号的小伙伴百分之百信任,奈何互联网鱼龙混杂,万一上传了什么不雅的图片,然后不巧被某部门发现了,我包括我的服务器域名可就彻底玩完了! 架构 如...

0
0
没有更多内容
加载失败,请刷新页面
点击加载更多
加载中
下一页
发表了问答
2019/01/25 18:43

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

回顾一周社区热门资讯与你的精彩留言~ 第【十二】期:0119-0125 一周一问,你有什么见解? 有没有一本书或教程,教你如何把一堆乱代码逐步优化成逻辑清晰,易读,复用性高的代码 来自:@OSC_NVnHth 站内需知 开源中国翻译频道贡献 2018 年 11 & 12 月奖励名单 精彩回复 from: Golang 1.11.5 和 1.10.8 版本发布 --- from...

2
4
没有更多内容
加载失败,请刷新页面
点击加载更多
加载中
下一页
1 评论
38 收藏
分享
OSCHINA
登录后可查看更多优质内容
返回顶部
顶部