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

项目简介

这是一个 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# 检测 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 次请求。