facebook
/

mask2former-swin-small-coco-instance

Image Segmentation

Model card Files Files and versions Community

adirik commited on Jan 4, 2023

Commit

d32b7ee

·

1 Parent(s): 120e87d

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 = 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)

 # 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)