项目简介
这是一个 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_64
、ARM64
。
使用 API 进行内容检查
curl -X POST -F "file=@/path/to/image.jpg" http://localhost:3333/check
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」公众号