facebook
/

mask2former-swin-base-coco-panoptic

Image Segmentation

Model card Files Files and versions

adirik commited on Jan 4, 2023

Commit

69472d8

·

1 Parent(s): dd96525

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 panoptic segmentation
-processor = Mask2FormerImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 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 panoptic segmentation
+processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"