概要
YOLOS smallは、Vision Transformerをベースにした物体検出モデルで、COCO 2017で学習済みの軽量版です。
画像内の物体クラスとバウンディングボックスを推定でき、PyTorchとHugging Face Transformersからそのまま利用できます。
DETR系の損失とHungarian matchingを使うシンプルな設計が特徴で、研究者や機械学習エンジニア、既存のPython推論パイプラインに物体検出を組み込みたい開発者向けです。
互換性・特徴
- Python
- PyTorch
- Transformers
- 物体検出
- GPU推奨
基本情報
| ライセンス | apache-2.0 |
| Likes | 94 |
| Downloads | 746,411 |
| Pipeline | object-detection |
| カテゴリ | その他 |
| アクティビティ | low |
HuggingFace: https://huggingface.co/hustvl/yolos-small