如何利用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