Celebrity Classifier

🚀 名人分類器

本項目提供的名人分類器模型，能夠將人臉準確分類至對應的名人。它基於特定數據集進行訓練和微調，在人臉分類任務中具有較高的準確性和實用性。

🚀 快速開始

模型描述

此模型可將人臉分類到對應的名人。它在 tonyassi/celebrity-1000 數據集上進行訓練，並在 google/vit-base-patch16-224-in21k 基礎上進行微調。

數據集描述

tonyassi/celebrity-1000 數據集包含排名前 1000 的名人，共有 18,184 張圖像，圖像尺寸為 256x256，且都經過裁剪以突出人臉。

如何使用

from transformers import pipeline

# 初始化圖像分類管道
pipe = pipeline("image-classification", model="tonyassi/celebrity-classifier")

# 進行分類
result = pipe('image.png')

# 打印結果
print(result)

📚 詳細文檔

訓練和評估數據

該模型在評估集上取得了以下結果：

• 損失值：0.9089
• 準確率：0.7982

訓練超參數

訓練過程中使用了以下超參數：

• 學習率：5e-05
• 訓練批次大小：16
• 評估批次大小：16
• 隨機種子：42
• 梯度累積步數：4
• 總訓練批次大小：64
• 優化器：Adam（β1=0.9，β2=0.999，ε=1e-08）
• 學習率調度器類型：線性
• 學習率調度器熱身比例：0.1
• 訓練輪數：20

框架版本

• Transformers 4.35.2
• Pytorch 2.1.0+cu121
• Datasets 2.16.1
• Tokenizers 0.15.0

📄 許可證

本項目採用 Apache-2.0 許可證。