facebook
/
mask2former-swin-base-coco-panoptic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Alara Dirik 2023-01-04 14:44:52 +00:00 committed by huggingface-web
parent dd965256b4
commit 69472d84a6
1 changed files with 2 additions and 2 deletions

4
README.md
View File

@ -40,11 +40,11 @@ Here is how to use this model:
import requests
import torch
from PIL import Image
from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegmentation
from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
# load Mask2Former fine-tuned on COCO panoptic segmentation
processor = Mask2FormerImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
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"