Vit Finetuned Chest Xray Pneumonia

🚀 vit-finetuned-chest-xray-pneumonia

該模型是 google/vit-base-patch16-224-in21k 在 chest-xray-pneumonia 數據集上的微調版本。它在評估集上取得了以下結果：

• 損失值：0.1271
• 準確率：0.9551

🚀 快速開始

本模型是基於預訓練的 google/vit-base-patch16-224-in21k 模型，在胸部X光肺炎數據集上進行微調得到的，可用於胸部X光圖像的肺炎分類任務。

🔧 技術細節

訓練超參數

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

• 學習率：2e-05
• 訓練批次大小：16
• 評估批次大小：8
• 隨機種子：42
• 優化器：Adam，其中 betas=(0.9, 0.999)，epsilon=1e-08
• 學習率調度器類型：線性
• 訓練輪數：10

訓練結果

💻 使用示例

示例圖片

肺炎胸部X光片

正常胸部X光片

框架版本

• Transformers 4.17.0
• Pytorch 1.10.0+cu111
• Datasets 1.18.4
• Tokenizers 0.11.6

📄 許可證

本模型採用 Apache-2.0 許可證。

⚠️ 重要提示

本README文檔是根據訓練器可訪問的信息自動生成的，你可能需要校對並完善它。