Views
No views yet
1from transformers import AutoImageProcessor, SegformerForSemanticSegmentation
2
3processor = AutoImageProcessor.from_pretrained("mnjm/segformer-b0-voc")
4model = SegformerForSemanticSegmentation.from_pretrained("mnjm/segformer-b0-voc")
5inputs = processor(images=image, return_tensors="pt")
6output = model(**inputs)