Segformer B5 Finetuned Human Parsing

🚀 segformer-b5-finetuned-human-parsing

該模型是基於 nvidia/mit-b5 在未知數據集上進行微調的版本，可用於圖像分割任務，在人體解析方面表現出色。

🚀 快速開始

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

• 損失值：0.2292
• 平均交併比（Mean Iou）：0.6258
• 平均準確率（Mean Accuracy）：0.7547
• 整體準確率（Overall Accuracy）：0.8256
• 背景準確率（Accuracy Background）：nan
• 帽子準確率（Accuracy Hat）：0.8561
• 頭髮準確率（Accuracy Hair）：0.8974
• 太陽鏡準確率（Accuracy Sunglasses）：0.7540
• 上衣準確率（Accuracy Upper - clothes）：0.8553
• 裙子準確率（Accuracy Skirt）：0.7026
• 褲子準確率（Accuracy Pants）：0.8913
• 連衣裙準確率（Accuracy Dress）：0.7525
• 腰帶準確率（Accuracy Belt）：0.4251
• 左腳鞋準確率（Accuracy Left - shoe）：0.6014
• 右腳鞋準確率（Accuracy Right - shoe）：0.6374
• 臉部準確率（Accuracy Face）：0.9094
• 左腿準確率（Accuracy Left - leg）：0.8452
• 右腿準確率（Accuracy Right - leg）：0.8343
• 左臂準確率（Accuracy Left - arm）：0.8506
• 右臂準確率（Accuracy Right - arm）：0.8287
• 包包準確率（Accuracy Bag）：0.8232
• 圍巾準確率（Accuracy Scarf）：0.3662
• 背景交併比（Iou Background）：0.0
• 帽子交併比（Iou Hat）：0.7625
• 頭髮交併比（Iou Hair）：0.8171
• 太陽鏡交併比（Iou Sunglasses）：0.6400
• 上衣交併比（Iou Upper - clothes）：0.7700
• 裙子交併比（Iou Skirt）：0.6211
• 褲子交併比（Iou Pants）：0.7788
• 連衣裙交併比（Iou Dress）：0.5512
• 腰帶交併比（Iou Belt）：0.3564
• 左腳鞋交併比（Iou Left - shoe）：0.5032
• 右腳鞋交併比（Iou Right - shoe）：0.5381
• 臉部交併比（Iou Face）：0.8294
• 左腿交併比（Iou Left - leg）：0.7412
• 右腿交併比（Iou Right - leg）：0.7591
• 左臂交併比（Iou Left - arm）：0.7579
• 右臂交併比（Iou Right - arm）：0.7705
• 包包交併比（Iou Bag）：0.7729
• 圍巾交併比（Iou Scarf）：0.2956

🔧 技術細節

訓練超參數

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

• 學習率（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）：15

訓練結果

📄 許可證

該項目使用其他許可證。