Atik16209 commited on
Commit
a5084bd
1 Parent(s): 46a205a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -35,10 +35,10 @@ Here is how to use this model:
35
 
36
  >>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
37
  >>> image = Image.open(requests.get(url, stream=True).raw)
38
- >>> feature_extractor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-base-ade")
39
  >>> inputs = feature_extractor(images=image, return_tensors="pt")
40
 
41
- >>> model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-base-ade")
42
  >>> outputs = model(**inputs)
43
  >>> # model predicts class_queries_logits of shape `(batch_size, num_queries)`
44
  >>> # and masks_queries_logits of shape `(batch_size, num_queries, height, width)`
 
35
 
36
  >>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
37
  >>> image = Image.open(requests.get(url, stream=True).raw)
38
+ >>> feature_extractor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-large-ade")
39
  >>> inputs = feature_extractor(images=image, return_tensors="pt")
40
 
41
+ >>> model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-ade")
42
  >>> outputs = model(**inputs)
43
  >>> # model predicts class_queries_logits of shape `(batch_size, num_queries)`
44
  >>> # and masks_queries_logits of shape `(batch_size, num_queries, height, width)`