bhky/opennsfw2

解决了什么问题

它提供了一个现代化、易于使用的 Yahoo Open-NSFW 模型实现，用于检测图像和视频中的色情内容，取代了已停止维护的基于 Caffe 的原始版本。

如何工作

该项目使用预训练 Yahoo 模型的 Keras 3 实现。它支持多种后端（TensorFlow 和 JAX），并为处理单张图像、图像批次和视频帧提供高级 API。它包含特定的预处理流水线（YAHOO 和 SIMPLE），以确保与原始模型的权重兼容。

适合谁

需要将 NSFW 内容检测集成到其应用程序中的开发人员，特别是那些使用 Keras/TensorFlow/JAX 生态系统的开发人员。

亮点

• 多后端支持：支持 Keras 3 (TensorFlow/JAX) 和 tf-keras。
• 视频分析：能够预测视频每一帧的 NSFW 概率。
• Grad-CAM 支持：生成热图以突出显示导致预测的图像区域（仅限 TensorFlow 后端）。
• 微调：提供使用 TensorFlow 在自定义数据集上训练或微调模型的示例和工具。