环境

• centos7
• docker
• caffe

运行

git clone git@github.com:yahoo/open_nsfw.git

wget http://180.76.143.82:3002/uploads/youyuan/1476149909738A7115F5.png

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 \
1476149909738A7115F5.png

报错

ImportError: No module named skimage.io

解决

docker run -it caffe:cpu
pip install -U scikit-image

重新运行caffe 发现依然报错，原因是容器安装skimage后没有commit到镜像

docker run -it --name caffe -d -it caffe:cpu
docker commit 38e56a9db75b caffe:cpu
docker rm -f 4e8724b11078
docker exec -it caffe bash
pip install -U scikit-image
docker commit caffe caffe:cpu

运行

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 \
1476149909738A7115F5.png

运行结果

NSFW score:   0.142687216401

参考

• open_nsfw
• Caffe搭建：常见问题解决办法和ubuntu使用中遇到问题（持续更新）
• 雅虎开源色情图片检测神经网络