1、服务端安装
把雅虎的鉴黄模型下载到服务器
git clone https://github.com/yahoo/open_nsfw.git
# 如果clone比较慢,可以直接下载zip包也行
2、构建Caffe Docker映像(CPU)
docker build -t caffe:cpu https://raw.githubusercontent.com/BVLC/caffe/master/docker/cpu/Dockerfile
3、进行鉴黄
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
# 图片要在open_nsfw目录里面
返回结果:
# 分数越高说明色情图的概率很大
NSFW score: 0.14057905972