nielsr HF staff shivi commited on
Commit
c8cf1b5
1 Parent(s): 0c9e54a

change image_processor_type and fix readme (#1)

Browse files

- change image_processor_type and fix readme (31d43a0cde07363427da33ec3a12138ee1e204bb)
- Update README.md (61adfa26c7e52abfecc262b6bee4581e0562daf2)


Co-authored-by: Shivalika Singh <shivi@users.noreply.huggingface.co>

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,