qubvel-org/segmentation_models.pytorch

Add Mask2Former to SMP

Open

#1 044 ouverte le 24 janv. 2025

Voir sur GitHub
 (5 commentaires) (1 réaction) (0 assignés)Python (1 835 forks)github user discovery
good difficult issuehelp wantednew-model

Métriques du dépôt

Stars
 (11 601 stars)
Métriques de merge PR
 (Métriques PR en attente)

Description

Mask2Former model was introduced in the paper Masked-attention Mask Transformer for Universal Image Segmentation and first released in this repository.

Mask2Former addresses instance, semantic and panoptic segmentation with the same paradigm: by predicting a set of masks and corresponding labels. Hence, all 3 tasks are treated as if they were instance segmentation. Mask2Former outperforms the previous SOTA, MaskFormer both in terms of performance an efficiency by (i) replacing the pixel decoder with a more advanced multi-scale deformable attention Transformer, (ii) adopting a Transformer decoder with masked attention to boost performance without without introducing additional computation and (iii) improving training efficiency by calculating the loss on subsampled points instead of whole masks.

Papers with Code https://paperswithcode.com/paper/masked-attention-mask-transformer-for

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

HF Reference implementation: https://huggingface.co/docs/transformers/main/en/model_doc/mask2former https://github.com/huggingface/transformers/blob/main/src/transformers/models/mask2former/modeling_mask2former.py

Guide contributeur