Hugging Face's logo

facebook
/
mask2former-swin-base-IN21k-ade-semantic

Image Segmentation
Transformers
PyTorch
mask2former
vision
Inference Endpoints
Model card Files Community
2
shivi commited on
Commit
7db3248
•
1 Parent(s): 6d26c9c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -43,7 +43,7 @@ from PIL import Image
43
  from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
44
 
45
 
46
- # load Mask2Former fine-tuned on COCO panoptic segmentation
47
  processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-IN21k-ade-semantic")
48
  model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-IN21k-ade-semantic")
49
 
 
43
  from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
44
 
45
 
46
+ # load Mask2Former fine-tuned on ADE20k semantic segmentation
47
  processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-IN21k-ade-semantic")
48
  model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-IN21k-ade-semantic")
49