一键部署 NSFW 检测模型:快速识别并过滤敏感图片内容

一、简介

  • nsfw_detector是一个 NSFW 内容检测器,支持快速docker私有部署,提供API服务
  • 低资源消耗,2GB内存即可运行该模型,多核CPU自动调度加速推理
  • 可以识别多种文件类型:图片、PDF、视频、压缩文件、Doc/Docx
  • 基于专门训练的 NSFW 模型识别,准确度较好
  • 该工具的开源地址:https://github.com/tmplink/nsfw_detector,使用的模型地址参考:https://huggingface.co/Falconsai/nsfw_image_detection
  • 该工具的架构设计/运行逻辑参考下图:

二、安装

  • 提前安装好docker、docker-compose软件环境
  • 新建docker-compose.yml文件,配置内容如下:
    version: '3'
    services:
      nsfw-detector:
        image: vxlink/nsfw_detector:latest
        container_name: nsfw-detector
        ports:
          - "3333:3333"
        restart: unless-stopped
  • 一键启动
    docker-compose up -d
  • 启动完成功之后,查看是否运行正常

三、使用示例

  1. 提供在线访问页面:http://127.0.0.1:3333/,可以在页面上传图片等内容进行识别,如下:

  2. 提供API,POST 接口:http://127.0.0.1:3333/check

    • Postman请求示例

    • 返回json结果,包含两个敏感度评判概率字段:normal(正常内容概率)、nsfw(敏感内容概率)。通常平台会设定一个阈值(如 nsfw 概率 > 0.5才拦截)

      {
          "filename": "PixPin_2025-05-13_08-55-39.png", //图片名称
          "result": {
              "normal": 0.48161378502845764, //表示图片被判定为正常内容的概率约为48.16%。
              "nsfw": 0.51838618516922 //表示图片被判定为敏感内容(NSFW)的概率约为51.84%。
          },
          "status": "success" //执行成功
      }

四、总结

  • 该工具部署快速简单、资源占用较低、专业模型加持准确度高,而且提供页面和接口,方便集成与使用
  • 可以对社交平台图片、视频等资源做内容初审,防止敏感内容传播造成不良影响
© 版权声明
THE END
喜欢就支持一下吧
点赞10 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容