← all repositories

NVlabs/SegFormer

A vision-transformer model for semantic segmentation that assigns class labels to every pixel in an image.

3.5k stars Python Computer Vision
SegFormer
Velocity · 7d
+1.9
★ / day
Trend
steady
star history

SegFormer is a semantic segmentation method using hierarchical transformers that was published at NeurIPS 2021. It provides official PyTorch training and evaluation code along with pretrained models. The approach uses a hierarchical transformer encoder combined with a lightweight all-MLP decoder head to achieve efficient pixel-level segmentation across datasets like ADE20K and Cityscapes.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.