facebook
/

mask2former-swin-small-coco-instance

Image Segmentation

Model card Files Files and versions

shivi commited on Jan 24, 2023

Commit

f77f179

·

1 Parent(s): f8eb176

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -44,8 +44,8 @@ from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegme
 # load Mask2Former fine-tuned on COCO instance segmentation
-processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-small-coco-instance")
-model = AutoForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-small-coco-instance")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

 # load Mask2Former fine-tuned on COCO instance segmentation
+processor = Mask2FormerImageProcessor.from_pretrained("facebook/mask2former-swin-small-coco-instance")
+model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-small-coco-instance")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)