Hugging Face BlogMar 17, 2022, 12:00 AM
Fine-Tune a Semantic Segmentation Model with a Custom Dataset
本教學詳細介紹如何使用 Hugging Face Transformers 微調 SegFormer 語義分割模型。
本指南介紹如何利用 Hugging Face 的 Transformers 庫微調 SegFormer 進行語義分割。內容涵蓋自定義資料集的準備、使用 SegformerImageProcessor 進行圖像預處理,以及設定 Trainer API 進行訓練。最後,教學展示了如何使用 mIoU 評估模型效能並進行推理。
想看英文原文 / 完整內容?
前往 Hugging Face Blog 原文 →摘要由 AI 整理,以原文為準。