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

Register feature extractor (#15634)

Browse Source 
* Rework AutoFeatureExtractor.from_pretrained internal

* Custom feature extractor

* Add more tests

* Add support for custom feature extractor code

* Clean up

* Add register API to AutoFeatureExtractor
This commit is contained in:
Sylvain Gugger
2022-02-14 13:35:16 -05:00
committed by GitHub
parent 0f71c29053
commit 2e11a04337
3 changed files with 56 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_feature_extraction_common.py
View File

@@ -43,6 +43,9 @@ if is_vision_available():
SAMPLE_FEATURE_EXTRACTION_CONFIG_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures")
SAMPLE_FEATURE_EXTRACTION_CONFIG_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures")
def prepare_image_inputs(feature_extract_tester, equal_resolution=False, numpify=False, torchify=False):
"""This function prepares a list of PIL images, or a list of numpy arrays if one specifies numpify=True,
or a list of PyTorch tensors if one specifies torchify=True.