基于Ubuntu下Docker安装使用Yahoo的NSFW

如何利用docker使用NSFW

1.  基于docker创建caffe 镜像（GPU）

$ cd open_nsfw
$ wget https://raw.githubusercontent.com/BVLC/caffe/master/docker/standalone/gpu/Dockerfile
$ docker build -t caffe:cpu ./

 创建只基于CPU模式的caffe镜像

$ wget https://raw.githubusercontent.com/BVLC/caffe/master/docker/standalone/cpu/Dockerfile

2. 检测caffe是否正确安装

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

3.  使用--volume参数，使得当前文件夹作为映射卷

docker run -ti --volume=$(pwd):/workspace caffe:cpu bash

进入the docker caffe image

root@8855e6188183:/workspace#

4. 在当前容器中运行nsfw 

root@8855e6188183:/workspace#python ./classify_nsfw.py --model_def nsfw_model/deploy.prototxt --pretrained_model nsfw_model/resnet_50_1by2_nsfw.caffemodel anderson_210.jpg

显示结果

NSFW score:   0.14057905972