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 <[email protected]>
- README.md +1 -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 |
-
|
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": "
|
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,
|