nsfwjs

NSFWJS 是一个功能强大的 JavaScript 库，旨在使用 TensorFlow.js 直接在客户端检测不雅或 NSFW（Not Safe For Work，工作场所不宜）图像。它允许开发者将图像分类集成到 Web 应用中，而无需将敏感图像发送到服务器。该库将图像分为五个不同的类别：Drawing（工作场所安全的绘图，包括动漫）、Neutral（工作场所安全的中性图像）、Porn（色情图像）、Sexy（性暗示但非色情）和 Hentai（色情绘图）。 主要特性包括： - **客户端执行**：完全在用户浏览器中运行，确保隐私并减少服务器负载。 - **高准确率**：小型模型准确率约 90%，中型模型约 93%。 - **多后端支持**：通过 TensorFlow.js 支持 WebGPU、WebGL、WASM 和 CPU 后端，实现跨浏览器兼容性和性能。 - **灵活的模型加载**：提供内置模型如 MobileNetV2、MobileNetV2Mid 和 InceptionV3，或允许用户托管自定义模型以减小打包体积。 - **简单的 API**：提供直观的方法，如 `load()` 初始化模型、`classify()` 分析图像（接受 `<img>`、`<video>` 或 `<canvas>` 元素）以及 `dispose()` 释放内存。 - **Node.js 支持**：也可使用 `@tensorflow/tfjs-node` 在服务端 Node.js 应用中使用。 该库采用 MIT 许可证，广泛用于开源项目，例如用于过滤 NSFW 内容的浏览器扩展。