Upload README.md with huggingface_hub
Browse files
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
|
45 |
processor = AutoImageProcessor.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2021-instance")
|
46 |
model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2021-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-2021-instance")
|
46 |
model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/video-mask2former-swin-small-youtubevis-2021-instance")
|
47 |
|