nielsr HF staff shivi commited on
Commit
cf8983c
1 Parent(s): 09e33d2

Update README and fix img processor type (#1)

Browse files

- Update README.md (950997e47af00faa7840bced1754928fc2c4a8a3)
- Update preprocessor_config.json (df7cafcb580f7fdd3668781d81eb53b831e03533)


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

Files changed (2) hide show
  1. README.md +1 -2
  2. preprocessor_config.json +1 -1
README.md CHANGED
@@ -60,9 +60,8 @@ class_queries_logits = outputs.class_queries_logits
60
  masks_queries_logits = outputs.masks_queries_logits
61
 
62
  # you can pass them to processor for postprocessing
63
- result = processor.post_process_semantic_segmentation(outputs, target_sizes=[image.size[::-1]])[0]
64
  # we refer to the demo notebooks for visualization (see "Resources" section in the Mask2Former docs)
65
- predicted_semantic_map = result["segmentation"]
66
  ```
67
 
68
  For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/mask2former).
 
60
  masks_queries_logits = outputs.masks_queries_logits
61
 
62
  # you can pass them to processor for postprocessing
63
+ predicted_semantic_map = processor.post_process_semantic_segmentation(outputs, target_sizes=[image.size[::-1]])[0]
64
  # we refer to the demo notebooks for visualization (see "Resources" section in the Mask2Former docs)
 
65
  ```
66
 
67
  For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/mask2former).
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,