diff --git a/README.md b/README.md index a97f908..185fd38 100644 --- a/README.md +++ b/README.md @@ -51,6 +51,7 @@ class_queries_logits = outputs.class_queries_logits masks_queries_logits = outputs.masks_queries_logits # you can pass them to feature_extractor for postprocessing +# we refer to the demo notebooks for visualization (see "Resources" section in the MaskFormer docs) predicted_semantic_map = feature_extractor.post_process_semantic_segmentation(outputs, target_sizes=[image.size[::-1]])[0] ```