facebook
/

mask2former-swin-tiny-coco-instance

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

adirik commited on Jan 4, 2023

Commit

afc8658

•

1 Parent(s): 74c8892

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -40,11 +40,11 @@ Here is how to use this model:
 import requests
 import torch
 from PIL import Image
-from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegmentation
 # load Mask2Former fine-tuned on COCO instance segmentation
-processor = Mask2FormerImageProcessor.from_pretrained("facebook/mask2former-swin-tiny-coco-instance")
 model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-tiny-coco-instance")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"

 import requests
 import torch
 from PIL import Image
+from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
 # load Mask2Former fine-tuned on COCO instance segmentation
+processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-tiny-coco-instance")
 model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-tiny-coco-instance")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"