https://github.com/yahoo/open_nsfw 地址先放上
首先在电脑上安装caffe
二选一,上面的更适合国内环境,基本上安装不会出问题。下面的是官方推荐的,安装失败几率很高 也可以自己安装caffe参考caffe安装方法
docker pull elezar/caffe:cpu
docker build -t caffe:cpu https://raw.githubusercontent.com/BVLC/caffe/master/docker/cpu/Dockerfile
查看是否安装成功
docker run -ti elezar/caffe:cpu caffe --version
下载项目本身
git clone https://github.com/yahoo/open_nsfw.git
test.jpg为测试图片
运行以下代码
docker run --volume=$(pwd):/workspace elezar/caffe:cpu python ./classify_nsfw.py --model_def nsfw_model/deploy.prototxt --pretrained_model nsfw_model/resnet_50_1by2_nsfw.caffemodel test.jpg
测试成功
评论区