nvidia
/

segformer-b5-finetuned-ade-640-640

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Feb 21, 2022

Commit

7993b42

•

1 Parent(s): c17441a

Update code example

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -35,8 +35,8 @@ from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmenta
 from PIL import Image
 import requests
-feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512")
-model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

 from PIL import Image
 import requests
+feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b5-finetuned-ade-512-512")
+model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b5-finetuned-ade-512-512")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)