起因:之前 ImgURL 一直使用雅虎的 Open NSFW 识别方案,但是 Open NSFW 这个项目 2019 年就归档且不再更新了。最近正在重构 ImgURL ,顺便准备寻找新的色情图像识别方案,结果找了一圈,没发现满足自己需求的,于是就手搓了一个,希望对开发者有帮助。

Github 开源:https://github.com/helloxz/nsfw
测试 API:https://nsfw.demo.mba/check?url=https://www.imgurl.org/static/images/logo.png
只需要把 URL 参数值替换为任意可访问的图片地址即可识别,个人感觉识别率比 Open NSFW 更强一些,而且速度也不错。(识别速度非 http 下载和请求速度)
介绍
基于 AdamCodd/vit-base-nsfw-detector/ 实现,并封装为 HTTP API 调用,可用于识别网络色情图像( nsfw 识别),适合纯 CPU 机器推理,轻量级实现方案。
特点
- 纯 CPU 推理,适合大多数云服务器或 VPS
- 支持的图片类型:jpg/png/bmp/webp
- 支持 HTTP API 调用
- 支持鉴权访问
- 轻量级实现,资源占用较低
- 识别速度较快
- 支持 Docker 私有部署
Docker 部署
docker run -d \
--name nsfw \
-p 6086:6086 \
--restart always \
helloz/nsfw
完整的说明可以访问 Github 查看: https://github.com/helloxz/nsfw

