用不可描述的图片做可以描述的事情

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

几天前看到 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

用不可描述的图片做可以描述的事情相关推荐

微信分享网页链接自定义图片和文字描述
今天开发需要微信分享网页链接自定义图片和文字描述,于是网上各种查找资料和各种看手册,遇到了好多坑,不过啃了半天以后终于把问题解决了,以防以后忘记,今天记录一下; 首先,由于微信已经发布了想要分享链接必 ...
微信自定义分享，网页分享微信实现标题，描述和图片的方法（2020）
公司做活动的,每次活动要让用户把活动H5页面分享到微信好友,朋友圈从而裂变,但是分享出来不显示描述和图标,很不正规.如下图: 所以想要实现能够自定义描述和图标的功能.如下图: 网上这方面的教程很多,但 ...
安卓微信二次分享不显示描述和图片
问题描述在做微信分享的时候有遇到过这样的问题,,第一次分享到微信,描述和图片都能正常显示,但是打开连接用微信自带的分享功能进行分享时,就会出现以下情况:描述变链接.图片不显示.标题变更. 导致的原因 ...
（100）详细描述一个你做过的项目，面试必问（二十四）（第20天）
(100)详细描述一个你做过的项目, 面试必问(二十四)(第20天) 1 文章目录 1)文章目录 2)FPGA初级课程介绍 3)FPGA初级课程架构 4)详细描述一个你做过的项目, 面试必问(二十四) ...
python自带的PIL库扩展图片大小给图片加上文字描述
利用python自带的PIL库扩展图片大小给图片加上文字描述.大多都是库函数调用,只是给定图片宽度后计算文字所需行数的代码需要写. 代码比较丑,but it works. #!/usr/bin/env ...
python add picture显示过大_利用Python自带PIL库扩展图片大小给图片加文字描述的方法示例...
前言最近的一个项目中需要在图片上添加文字,使用了OpenCV,结果发现利用opencv给图像添加文字有局限.可利用的字体类型比较少,需要安装Freetype扩展,比较复杂.而且不能用putText函 ...
微信分享网页时自定义标题、描述和图片
用微信打开一个网页,选择右上角的"发送给朋友"后,收到的消息是这样的: 而为了推广效果,我们更希望能自定义标题.描述和图片,效果如下图: 下面介绍下怎样来实现这个功能 <sc ...
微信内置浏览器第三方网页分享链接给朋友以及转发到朋友圈无链接图片和链接描述的问题
解决问题:微信内置浏览器第三方网页分享链接给朋友以及转发到朋友圈无链接图片和链接描述的问题记录时间:2018-11-07 分享作者:小鹏谨记:一定要按步骤一步一步来先写一写思路:进入页面需要分享 ...
html分享标题描述图片尺寸,微信分享网页时自定义标题、描述和图片
用微信打开一个网页,选择右上角的"发送给朋友"后,收到的消息是这样的: 而为了推广效果,我们更希望能自定义标题.描述和图片,效果如下图: 实现工具:GetwxLink 联系QQ:6 ...

用不可描述的图片做可以描述的事情

用不可描述的图片做可以描述的事情

数据

使用

模型

用不可描述的图片做可以描述的事情相关推荐

最新文章

热门文章