Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -59,7 +59,7 @@ class_queries_logits = outputs.class_queries_logits
|
|
59 |
masks_queries_logits = outputs.masks_queries_logits
|
60 |
|
61 |
# you can pass them to processor for postprocessing
|
62 |
-
result =
|
63 |
# we refer to the demo notebooks for visualization (see "Resources" section in the Mask2Former docs)
|
64 |
predicted_video_instance_map = result["segmentation"]
|
65 |
```
|
|
|
59 |
masks_queries_logits = outputs.masks_queries_logits
|
60 |
|
61 |
# you can pass them to processor for postprocessing
|
62 |
+
result = image_processor.post_process_video_instance_segmentation(outputs, target_sizes=[tuple(video.shape[1:3])])[0]
|
63 |
# we refer to the demo notebooks for visualization (see "Resources" section in the Mask2Former docs)
|
64 |
predicted_video_instance_map = result["segmentation"]
|
65 |
```
|