概要
SegFormer B2をATRベースの人間パースデータセットで衣服セグメンテーション向けに微調整したモデルです。
人物画像から帽子、上着、スカート、パンツ、靴、バッグなど18分類で領域分割でき、人全体のセグメンテーションにも活用できます。
Transformers経由で推論できるPython実装例と評価指標が示されており、ファッション解析、前処理、自作アプリへの組み込みを行いたい機械学習エンジニアや研究用途の利用者向けです。
互換性・特徴
- Python
- Transformers
- 画像セグメンテーション
- GPU推奨
- CLI
基本情報
| ライセンス | other |
| Likes | 499 |
| Downloads | 225,450 |
| Pipeline | image-segmentation |
| カテゴリ | その他 |
| アクティビティ | mid |
HuggingFace: https://huggingface.co/mattmdjaga/segformer_b2_clothes