鉴黄-深度学习Caffe模型之open_nsfw

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

NSFW( Not Suitable for Work )– “不适合上班时观看”。

项目地址：open_nsfw

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

• 安装Docker：Ubuntu 16.04 安装 docker

• 下载open_nsfw

$ git clone https://github.com/yahoo/open_nsfw

• Build a caffe docker image (CPU)
  Note：git的地址有点错误，要去掉standalone; 如果是gpu版本，把cpu改成gpu即可。
  下面这句命令比较久，看网络情况，公司网络太慢，我花了>3个小时。

docker build -t caffe:cpu https://raw.githubusercontent.com/BVLC/caffe/master/docker/cpu/Dockerfile

• 检查caffe 的版本号

docker run caffe:cpu caffe --version
caffe version 1.0.0-rc3

• 启动Docker，并映射工作目录（open_nsfw），测试图片test_image.jpg
  （图片的读写权限要设置成可读）

cd open_nsfw
docker run --volume=$(pwd):/workspace caffe:cpu \
python ./classify_nsfw.py \
--model_def nsfw_model/deploy.prototxt \
--pretrained_model nsfw_model/resnet_50_1by2_nsfw.caffemodel \
test_image.jpg

---

测试1：#

1.jpg

NSFW 得分: 0.0429051183164##

---

测试2：#

5.jpg

NSFW 得分:0.941415905952##

---

测试3：#

4.jpg

NSFW 得分: 0.669750630856##