open_nsfw: 基于Caffe的成人图片识别模型

open_nsfw是雅虎开发的,基于Caffe深度神经网络训练的成人/淫秽图片分类模型。

官方使用 Not Suitable for Work – “不适合上班时观看”。项目地址::https://github.com/yahoo/open_nsfw

输入要检测的图片,它会返回图片评级(0-1),等级越高,图片越黄越暴力。你可以使用它过滤掉尺度较大的图像。

安装测试(Ubuntu)

open_nsfw是基于Caffe训练的模型,所以我们需要安装Caffe。为了简单,我使用现成的Caffa Docker容器。

下载open_nsfw:

安装Docker:Ubuntu 16.04 安装 docker

构建Caffe容器:

GPU版本:

CPU版本:

查看caffe版本:

启动Docker,并映射工作目录(open_nsfw):

测试:test1.jpg:

%e5%b1%8f%e5%b9%95%e5%bf%ab%e7%85%a7-2016-10-06-%e4%b8%8b%e5%8d%888-06-34

结果:

测试:test2.jpg:

345e5e57-eec7-4df4-895e-7ed787c923a81

结果:

结果大于0.8就非常”不适合上班时观看”,上图已经达到了0.99+,嘿嘿。

大片分类模型:https://github.com/ryanjay0/miles-deep-Deep Learning Porn Video Classifier/Editor with Caffe

相关文章

发表评论

电子邮件地址不会被公开。 必填项已用*标注