图像不适宜内容检测工具NSFW Detector

文摘   2024-11-18 10:16   湖南  

项目简介

这是一个 NSFW 内容检测器,它基于 Falconsai/nsfw_image_detection 。
模型: google/vit-base-patch16-224-in21k

你可以在这里验证它的准确度:NSFW 检测器

相比其它常见的 NSFW 检测器,这个检测器的优势在于:

  • 基于 AI ,准确度更好。

  • 支持纯 CPU 推理,可以运行在大部分服务器上。

  • 自动调度多个 CPU 加速推理。

  • 简单判断,只有两个类别:nsfw 和 normal。

  • 以 API 的方式提供服务,更方便集成到其它应用中。

  • 基于 Docker 部署,便于分布式部署。

  • 纯本地,保护您的数据安全。


性能需求

运行这个模型最多需要 2GB 的内存。不需要显卡的支持。
在同时处理大量请求时,可能需要更多的内存。


支持的文件类型

这个检测器支持检查的文件类型:

  • ✅ 图片(已支持)

  • ✅ PDF 文件(已支持)

  • ✅ 视频(已支持)

  • ⏳ 压缩包中的文件(计划)


快速开始

启动 API 服务器

docker run -d -p 3333:3333 --name nsfw-detector vxlink/nsfw_detector:latest

支持的系统架构:x86_64ARM64


使用 API 进行内容检查

# 检测图片curl -X POST -F "file=@/path/to/image.jpg" http://localhost:3333/check# 检测 PDF 文件curl -X POST -F "file=@/path/to/file.pdf" http://localhost:3333/pdf# 检测视频文件curl -X POST -F "file=@/path/to/file.mp4" http://localhost:3333/video

公共 API

可以使用 vx.link 提供的公共 API 服务来检测 NSFW 内容。

# 检测文件,会自动识别文件类型curl -X POST -F "file=@/path/to/image.jpg" https://vx.link/public/nsfw
  • 不会保存你提交的图片。

  • 请注意,该 API 速率限制为每分钟 30 次请求。




项目链接

https://github.com/tmplink/nsfw_detector/blob/main/README_cn.md

扫码加入技术交流群,备注开发语言-城市-昵称

合作请注明


 

关注「GitHubStore」公众号


GitHubStore
分享有意思的开源项目
 最新文章