Vit Base Patch16 224 In21k Male Or Female Eyes

🚀 vit-base-patch16-224-in21k_male_or_female_eyes

このモデルは、google/vit-base-patch16-224-in21k を imagefolder データセットでファインチューニングしたものです。評価セットでは以下の結果を達成しています。

• 損失: 0.0810
• 正解率: 0.9727
• F1値: 0.9741
• 再現率: 0.9666
• 適合率: 0.9818

📚 ドキュメント

モデルの説明

これは男性と女性の目を区別する二値分類モデルです。

作成方法の詳細については、次のリンクを参照してください。https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Image%20Classification/Binary%20Classification/Male%20or%20Female%20Eyes/are_they_male_or_female_eyes_ViT.ipynb

想定される用途と制限

このモデルは、技術を用いて複雑な問題を解決する能力を示すために作成されました。

学習と評価データ

データセットのソース: https://www.kaggle.com/datasets/pavelbiz/eyes-rtte

データセットのサンプル画像:

学習手順

学習ハイパーパラメータ

学習中に使用されたハイパーパラメータは以下の通りです。

• 学習率: 0.0002
• 学習バッチサイズ: 16
• 評価バッチサイズ: 8
• 乱数シード: 42
• オプティマイザ: Adam (betas=(0.9,0.999), epsilon=1e-08)
• 学習率スケジューラ: 線形
• エポック数: 3

学習結果

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

• Transformers 4.25.1
• Pytorch 1.12.1
• Datasets 2.8.0
• Tokenizers 0.12.1

📄 ライセンス

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

モデル情報