Segformer B0 Finetuned Segments Water 2

🚀 segformer-b0-finetuned-segments-water-2

該模型是在圖像分割領域，基於特定數據集對預訓練模型進行微調的成果，可用於圖像分割相關任務。

🚀 快速開始

此模型是 nvidia/mit-b0 在 imadd/water_dataset 數據集上的微調版本。它在評估集上取得了以下結果：

• 損失值：0.5845
• 平均交併比（Mean Iou）：nan
• 平均準確率（Mean Accuracy）：nan
• 總體準確率（Overall Accuracy）：nan
• 每類交併比（Per Category Iou）：[nan, nan]
• 每類準確率（Per Category Accuracy）：[nan, nan]

🔧 技術細節

訓練超參數

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

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

訓練結果

框架版本

• Transformers 4.20.1
• Pytorch 1.11.0 + cu113
• Datasets 2.3.2
• Tokenizers 0.12.1

📄 許可證

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