Add post_process_semantic_segmentation method to DPTFeatureExtractor (#19107)

* add post-processing method for semantic segmentation * add test for post-processing
2022-09-21 15:15:26 +03:00
parent da6a1b6ca1
commit e7fdfc720a
3 changed files with 68 additions and 3 deletions
--- a/docs/source/en/model_doc/dpt.mdx
+++ b/docs/source/en/model_doc/dpt.mdx
@@ -37,6 +37,7 @@ This model was contributed by [nielsr](https://huggingface.co/nielsr). The origi

 [[autodoc]] DPTFeatureExtractor
    - __call__
+    - post_process_semantic_segmentation


 ## DPTModel