侧边栏壁纸
  • 累计撰写 38 篇文章
  • 累计创建 8 个标签
  • 累计收到 3 条评论

目 录CONTENT

文章目录

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

Administrator
2022-06-16 / 0 评论 / 2 点赞 / 589 阅读 / 0 字
温馨提示:
部分素材来自网络,若不小心影响到您的利益,请联系我们删除。

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
  1. 支付宝打赏

    qrcode alipay
  2. 微信打赏

    qrcode weixin

评论区