Nsfw Image Detector

🚀 NSFW画像検出器

このモデルは画像分類を行うためのモデルで、google/vit-base-patch16-224-in21k をベースにfine-tuningされており、高い精度で画像の分類が可能です。

🚀 クイックスタート

このモデルは google/vit-base-patch16-224-in21k を imagefolder データセットでfine-tuningしたバージョンです。評価セットでは以下の結果を達成しています。

• 損失: 0.8138
• 正解率: 0.9316
• 正解率 K: 0.9887

384バージョンには以下のリンクからアクセスできます。

https://huggingface.co/LukeJacob2023/nsfw-image-detector-384

📚 ドキュメント

モデルの説明

ラベル: ['drawings', 'hentai', 'neutral', 'porn', 'sexy']

想定される用途と制限

学習と評価データ

約28,000枚の画像を含むカスタムデータセットを使用しています。もしあなたのドメインの精度を向上させたい場合は、データセットを提供していただければ助かります。

学習手順

学習ハイパーパラメータ

学習中に以下のハイパーパラメータが使用されました。

• 学習率: 2e-05
• 学習バッチサイズ: 32
• 評価バッチサイズ: 32
• シード: 42
• オプティマイザ: Adam (betas=(0.9,0.999), epsilon=1e-08)
• 学習率スケジューラの種類: linear
• 学習率スケジューラのウォームアップステップ: 500
• エポック数: 10
• 混合精度学習: Native AMP

学習結果

フレームワークのバージョン

• Transformers 4.36.2
• Pytorch 2.0.0
• Datasets 2.15.0
• Tokenizers 0.15.0

📄 ライセンス

このモデルは Apache-2.0 ライセンスの下で提供されています。