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