segformer

tiatoolbox . models . architecture . segformer

SegFormer semantic segmentation via Hugging Face Transformers (Apache-2.0).

This module wraps transformers.SegformerForSemanticSegmentation and keeps the TIAToolbox ModelABC inference API. SMP-format checkpoints are auto-remapped on load_state_dict.

Classes

Segformer

SegFormer semantic segmentation model (Hugging Face Transformers).