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-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 |
|