nielsr HF staff shivi commited on
Commit
dd3578c
1 Parent(s): f8eb176

Update README.md (#1)

Browse files

- Update README.md (f77f179fc62b98f6734481ebbef5bceeb1e7f10b)
- Update README.md (810b5a1b4661b9ee6f4e8ba1e4e752fa8dd836be)
- Update preprocessor_config.json (e31996f7b8e731b0131a4ae13d5a2f283480d458)


Co-authored-by: Shivalika Singh <[email protected]>

Files changed (2) hide show
  1. README.md +2 -2
  2. preprocessor_config.json +1 -1
README.md CHANGED
@@ -40,12 +40,12 @@ Here is how to use this model:
40
  import requests
41
  import torch
42
  from PIL import Image
43
- from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegmentation
44
 
45
 
46
  # load Mask2Former fine-tuned on COCO instance segmentation
47
  processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-small-coco-instance")
48
- model = AutoForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-small-coco-instance")
49
 
50
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
51
  image = Image.open(requests.get(url, stream=True).raw)
 
40
  import requests
41
  import torch
42
  from PIL import Image
43
+ from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
44
 
45
 
46
  # load Mask2Former fine-tuned on COCO instance segmentation
47
  processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-small-coco-instance")
48
+ model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-small-coco-instance")
49
 
50
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
51
  image = Image.open(requests.get(url, stream=True).raw)
preprocessor_config.json CHANGED
@@ -9,7 +9,7 @@
9
  0.4560000002384186,
10
  0.4059999883174896
11
  ],
12
- "image_processor_type": "MaskFormerImageProcessor",
13
  "image_std": [
14
  0.2290000021457672,
15
  0.2239999920129776,
 
9
  0.4560000002384186,
10
  0.4059999883174896
11
  ],
12
+ "image_processor_type": "Mask2FormerImageProcessor",
13
  "image_std": [
14
  0.2290000021457672,
15
  0.2239999920129776,