shivi commited on
Commit
8de5656
1 Parent(s): d054974

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -41,7 +41,7 @@ from huggingface_hub import hf_hub_download
41
  from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
42
 
43
 
44
- # load Mask2Former fine-tuned on COCO instance segmentation
45
  processor = AutoImageProcessor.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2019-instance")
46
  model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2019-instance")
47
 
 
41
  from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
42
 
43
 
44
+ # load Mask2Former trained on YouTubeVIS 2021 instance segmentation
45
  processor = AutoImageProcessor.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2019-instance")
46
  model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2019-instance")
47