mask2former_swin_tiny¶

lucid.models.mask2former_swin_tiny(num_labels: int, *, pretrained_backbone: bool = False, **config_kwargs: Any) → Mask2Former¶

The mask2former_swin_tiny function builds a Mask2Former model with a Swin-Tiny backbone preset.

Total Parameters (num_labels=150): 47,439,633

Function Signature¶

@register_model
def mask2former_swin_tiny(
    num_labels: int,
    *,
    pretrained_backbone: bool = False,
    **config_kwargs
) -> Mask2Former

Parameters¶

num_labels (int): Number of segmentation classes.
pretrained_backbone (bool, optional): If True, initializes the backbone from available pretrained weights.
config_kwargs (dict, optional): Additional overrides applied to Mask2FormerConfig.

Returns¶

Mask2Former: Mask2Former model configured with a Swin-Tiny backbone.

Example Usage¶

from lucid.models.vision.mask2former import mask2former_swin_tiny
import lucid

model = mask2former_swin_tiny(num_labels=150)
x = lucid.random.randn(1, 3, 224, 224)
pred = model.predict(x)
print(pred.shape)

Pretrained Weights¶

import lucid.models as models
import lucid.weights as W

model = models.mask2former_swin_tiny(
    num_labels=150,
    weights=W.Mask2Former_Swin_Tiny_Weights.ADE20K_SEMANTIC,
)