鉴黄-深度学习Caffe模型之open_nsfw
2017-05-17 本文已影响555人
南风无影
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
