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

Add ImageFeatureExtractionMixin (#10905)

Browse Source 
* Add ImageFeatureExtractionMixin

* Add dummy vision objects

* Add require_vision

* Add tests

* Fix test
This commit is contained in:
Sylvain Gugger
2021-03-26 11:23:56 -04:00
committed by GitHub
parent 3c27d246e5
commit b0595d33c1
8 changed files with 544 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docs/source/main_classes/feature_extractor.rst
View File

@@ -39,3 +39,10 @@ BatchFeature
.. autoclass:: transformers.BatchFeature
:members:
ImageFeatureExtractionMixin
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.image_utils.ImageFeatureExtractionMixin
:members: