optimum
/

segformer-b0-finetuned-ade-512-512

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

Jingya HF staff commited on Dec 14, 2022

Commit

bc01a2c

•

1 Parent(s): 385e58b

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -56,6 +56,7 @@ from optimum.onnxruntime import ORTModelForSemanticSegmentation
 image_processor = SegformerImageProcessor.from_pretrained("optimum/segformer-b0-finetuned-ade-512-512")
 model = ORTModelForSemanticSegmentation.from_pretrained("optimum/segformer-b0-finetuned-ade-512-512")
 pipe = pipeline("image-segmentation", model=model, feature_extractor=image_processor)
 pred = pipe(url)
 ```

 image_processor = SegformerImageProcessor.from_pretrained("optimum/segformer-b0-finetuned-ade-512-512")
 model = ORTModelForSemanticSegmentation.from_pretrained("optimum/segformer-b0-finetuned-ade-512-512")
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 pipe = pipeline("image-segmentation", model=model, feature_extractor=image_processor)
 pred = pipe(url)
 ```