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
网友评论