Add post_process_semantic_segmentation method to SegFormer (#19072)

* add post_process_semantic_segmentation method to SegformerFeatureExtractor * add test for semantic segmentation post-processing
2022-09-21 11:40:35 +03:00
parent ef6741fe65
commit 9e95706648
3 changed files with 75 additions and 2 deletions
--- a/docs/source/en/model_doc/segformer.mdx
+++ b/docs/source/en/model_doc/segformer.mdx
@@ -93,6 +93,7 @@ SegFormer's results on the segmentation datasets like ADE20k, refer to the [pape

 [[autodoc]] SegformerFeatureExtractor
    - __call__
+    - post_process_semantic_segmentation

 ## SegformerModel