深度学习Caffe

http://www.jianshu.com/p/e746454958a3
鉴黄-深度学习Caffe模型之open_nsfw
96 作者 Gongjia 关注
2017.05.17 09:37* 字数 327 阅读 187评论 0喜欢 1
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