Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • NielsRogge's avatar
    Improve semantic segmentation models (#14355) · a2864a50
    NielsRogge authored 
    * Improve tests

* Improve documentation

* Add ignore_index attribute

* Add semantic_ignore_index to BEiT model

* Add segmentation maps argument to BEiTFeatureExtractor

* Simplify SegformerFeatureExtractor and corresponding tests

* Improve tests

* Apply suggestions from code review

* Minor docs improvements

* Streamline segmentation map tests of SegFormer and BEiT

* Improve reduce_labels docs and test

* Fix code quality

* Fix code quality again
    a2864a50
test_feature_extraction_beit.py 12.3 KB