开源一个色情图像(NSFW)识别方案,支持私有部署和 HTTP 调用

查看 26|回复 2
作者:xiaoz   
起因:之前 ImgURL 一直使用雅虎的 Open NSFW 识别方案,但是 Open NSFW 这个项目 2019 年就归档且不再更新了。最近正在重构 ImgURL ,顺便准备寻找新的色情图像识别方案,结果找了一圈,没发现满足自己需求的,于是就手搓了一个,希望对开发者有帮助。

Github 开源:https://github.com/helloxz/nsfw
测试 API:https://nsfw.demo.mba/check?url=https://www.imgurl.org/static/images/logo.png
只需要把 URL 参数值替换为任意可访问的图片地址即可识别,个人感觉识别率比 Open NSFW 更强一些,而且速度也不错。(识别速度非 http 下载和请求速度)
介绍
基于 AdamCodd/vit-base-nsfw-detector/ 实现,并封装为 HTTP API 调用,可用于识别网络色情图像( nsfw 识别),适合纯 CPU 机器推理,轻量级实现方案。
特点
  • 纯 CPU 推理,适合大多数云服务器或 VPS
  • 支持的图片类型:jpg/png/bmp/webp
  • 支持 HTTP API 调用
  • 支持鉴权访问
  • 轻量级实现,资源占用较低
  • 识别速度较快
  • 支持 Docker 私有部署

    Docker 部署
    docker run -d \
      --name nsfw \
      -p 6086:6086 \
      --restart always \
      helloz/nsfw
    完整的说明可以访问 Github 查看: https://github.com/helloxz/nsfw

    色情图像识别, nsfw, HTTP, API

  • MIUIOS   
    可以看下数据集吗,我需要人工详细标注一下

    xiaoz
    OP
      
    @MIUIOS #1,模型用的是 https://huggingface.co/AdamCodd/vit-base-nsfw-detector/ 开源的,我本身并没有进行模型训练。
    所以我这里没有数据集,哈哈。
    您需要登录后才可以回帖 登录 | 立即注册

    返回顶部