Vit Fire Detection

🚀 火災檢測視覺變換器（vit-fire-detection）

本模型是一個用於火災檢測的視覺變換器模型，它基於預訓練模型進行微調，在評估集上取得了高精度和高召回率的良好效果，可有效應用於火災檢測場景。

🚀 快速開始

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

• 損失值：0.0126
• 精確率：0.9960
• 召回率：0.9960

📚 詳細文檔

訓練和評估數據

相關信息待補充。

模型描述

相關信息待補充。

預期用途與限制

相關信息待補充。

🔧 技術細節

訓練過程

訓練超參數

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

• 學習率：0.0002
• 訓練批次大小：32
• 評估批次大小：32
• 隨機種子：42
• 優化器：Adam（β1 = 0.9，β2 = 0.999，ε = 1e-08）
• 學習率調度器類型：線性
• 學習率調度器預熱步數：100
• 訓練輪數：10

訓練結果

框架版本

• Transformers 4.25.1
• Pytorch 1.14.0.dev20221111
• Datasets 2.8.0
• Tokenizers 0.12.1

📄 許可證

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