nvidia
/

segformer-b0-finetuned-ade-512-512

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Apr 24, 2023

Commit

c30f696

·

1 Parent(s): 6010ab2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -31,11 +31,11 @@ You can use the raw model for semantic segmentation. See the [model hub](https:/
 Here is how to use this model to classify an image of the COCO 2017 dataset into one of the 1,000 ImageNet classes:
 ```python
-from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 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"

 Here is how to use this model to classify an image of the COCO 2017 dataset into one of the 1,000 ImageNet classes:
 ```python
+from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
 from PIL import Image
 import requests
+processor = SegformerImageProcessor.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"