SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add Universal Segmentation class + mapping (#20766)

Browse Source 
* Add mapping

* Add mapping to pipeline

* Apply suggestions

* Fix feature extractor tests

* Use ForInstance, add model to universal mapping

* More fixes

* Remove model from deprecated objectsé

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
This commit is contained in:
NielsRogge
2022-12-16 14:22:46 +01:00
committed by GitHub
parent e65445b4d6
commit 7f99861218
7 changed files with 45 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/en/model_doc/auto.mdx
View File

@@ -254,6 +254,10 @@ The following auto classes are available for the following computer vision tasks
[[autodoc]] AutoModelForInstanceSegmentation
### AutoModelForUniversalSegmentation
[[autodoc]] AutoModelForUniversalSegmentation
### AutoModelForZeroShotObjectDetection
[[autodoc]] AutoModelForZeroShotObjectDetection