oneformer_ade20k_swin_largeModel41/100 via “ade20k-150-class-semantic-prediction”
image-segmentation model by undefined. 1,02,623 downloads.
Unique: Trained on ADE20K's diverse 150-class taxonomy covering both stuff (wall, sky, floor) and things (person, car, furniture) with class-balanced sampling during training. Uses learned class embeddings (150×256) that are matched against pixel features via dot-product attention, enabling efficient per-pixel classification.
vs others: Achieves 48.9 mIoU on ADE20K validation set, outperforming DeepLabV3+ (46.2 mIoU) and comparable to Mask2Former (48.7 mIoU) while using a unified architecture. However, task-specific semantic segmentation models (e.g., SegFormer) can achieve 50+ mIoU if not constrained to multi-task design.