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 を使用してカスタムデータセットでモデルをトレーニングまたはファインチューニングするための例とツールを提供します。