侧边栏壁纸
  • 累计撰写 31 篇文章
  • 累计创建 6 个标签
  • 累计收到 2 条评论
标签搜索

目 录CONTENT

文章目录

使用开源项目完成图片的智能鉴黄 yahoo/open_nsfw

lj2508_com
2022-06-16 / 0 评论 / 2 点赞 / 330 阅读 / 176 字

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


image-1656212317518
查看是否安装成功

docker run -ti elezar/caffe:cpu caffe --version

image-1656212683823
下载项目本身

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

image-1656212750440
test.jpg为测试图片
image-1656233745610
运行以下代码

 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

image-1656233788586
测试成功

2

评论区