跳到正文
原文
Hugging Face Blog·· 2022-03-17AI 评分35

使用自定义数据集微调 SegFormer 语义分割模型

Fine-Tune a Semantic Segmentation Model with a Custom Dataset

AI 导读

本指南演示如何使用 🤗 transformers 库微调 SegFormer 模型,以构建披萨配送机器人的视觉系统。通过加载 segments/sidewalk-semantic 数据集并应用 ColorJitter 数据增强,模型能精准识别人行道像素及障碍物。

来源:Hugging Face Blog · huggingface.co