mnjm/segformer-b0-voc

segformer-bo-voc

Segormer b0 model trained from scratch on VoC 2007 and 2012 dataset using a custom training repo

Eval results can be found here

Run

from transformers import AutoImageProcessor, SegformerForSemanticSegmentation

processor = AutoImageProcessor.from_pretrained("mnjm/segformer-b0-voc")
model = SegformerForSemanticSegmentation.from_pretrained("mnjm/segformer-b0-voc")
inputs = processor(images=image, return_tensors="pt")
output = model(**inputs)