qubvel-org/segmentation_models.pytorch

Add Beit segmentation model

Open

#1024 aperta il 7 gen 2025

Vedi su GitHub
 (10 commenti) (1 reazione) (0 assegnatari)Python (1835 fork)github user discovery
good first issuehelp wantednew-model

Metriche repository

Star
 (11.601 star)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

Add Beit to SMP

BEiT-3 is a general-purpose multimodal foundation model developed by Microsoft that excels in various vision and vision-language tasks, including semantic segmentation. It employs a unified architecture with Multiway Transformers, enabling both deep fusion and modality-specific encoding. Pretrained using a masked "language" modeling approach on images ("Imglish"), texts, and image-text pairs, BEiT-3 effectively models images as another language. This design allows it to achieve state-of-the-art performance across a wide range of tasks, such as object detection, image classification, and semantic segmentation.

  • Achieves top 1 results on ADE20K-val

Papers with Code: https://paperswithcode.com/paper/image-as-a-foreign-language-beit-pretraining

Paper: https://arxiv.org/abs/2208.10442

HF reference implementation: https://huggingface.co/docs/transformers/model_doc/beit https://github.com/huggingface/transformers/blob/v4.47.1/src/transformers/models/beit/modeling_beit.py

Comments

As an example pls see the latest model additions:

Guida contributor