红联Linux门户
Linux帮助

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

发布时间:2017-04-21 11:42:55来源:linux网站作者:老吴的私房菜
如何利用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
 
本文永久更新地址:http://www.linuxdiyf.com/linux/30195.html