Add semantic segmentation post-processing method to MobileViT (#19105)

* add post-processing method for semantic segmentation Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-09-23 16:24:28 +03:00
parent 905635f5d3
commit 7e84723fe4
3 changed files with 73 additions and 2 deletions
--- a/docs/source/en/model_doc/mobilevit.mdx
+++ b/docs/source/en/model_doc/mobilevit.mdx
@@ -66,6 +66,7 @@ This model was contributed by [matthijs](https://huggingface.co/Matthijs). The T

 [[autodoc]] MobileViTFeatureExtractor
    - __call__
+    - post_process_semantic_segmentation

 ## MobileViTModel