大家好,这里是 NSFW 检测项目。
近期,为了提供更好的体验,我们为钛盘增加了 NSFW 检测项目,以便在提供预览功能时,能够自动使用替代图案显示那些 NSFW 图片。为此,我们尝试了许多开源项目,并最终建立了这个项目。
这个项目挑选了其中检测效果和性能最好的一个模型,它基于 Falconsai/nsfw_image_detection 提供的训练结果来向用户提供 API 服务。
其中,模型为:google/vit-base-patch16-224-in21k
特性一览:
我们目前正在打包相关代码以制作成 Docker 镜像,方便大家一键部署。预计本周内上线。
这个 Docker 镜像将会托管到 DockerHub,构建镜像的代码托管到 Github 上。
以下是 Docker 镜像要实现的一些特性:
- 支持 CPU 推理检测(会利用多核心)
- 内存占用在 2GB 以下(最大值)
- 响应迅速
- 通过 POST 传入图片进行检测,每次处理一张图片
现在,您可以在我们提供的公共 API 上来体验检测效果,我们不会留存您提交的图片。
API 地址 :https://vx.link/public/nsfw
示例代码:curl -X POST -F "file=@test.png" https://vx.link/public/nsfw
速率限制:每分钟 30 次请求
返回结果:{"data":{"nsfw":0.9909670948982239,"normal":0.010517489165067673},"status":1}
本项目以 Apache 2.0 许可证发布。