transformers/tests/pipelines
NielsRogge 4973d2a04c
Add Audio Spectogram Transformer (#19981)
* First draft

* Make conversion script work

* Add id2label mapping, run code quality

* Fix copies

* Add first draft of feature extractor

* Update conversion script to use feature extractor

* Make more tests pass

* Add docs

* update input_features to input_values + pad by default to max length

* Fix doc tests

* Add feature extractor tests

* Add proper padding/truncation to feature extractor

* Add support for conversion of all audioset checkpoints

* Improve docs and extend conversion script

* Fix README

* Rename spectogram to spectrogram

* Fix copies

* Add integration test

* Remove dummy conv

* Update to ast

* Update organization

* Fix init

* Rename model to AST

* Add require_torchaudio annotator

* Move import of ASTFeatureExtractor under a is_speech_available

* Fix rebase

* Add pipeline config

* Update name of classifier head

* Rename time_dimension and frequency_dimension for clarity

* Remove print statement

* Fix pipeline test

* Fix pipeline test

* Fix index table

* Fix init

* Fix conversion script

* Rename to ForAudioClassification

* Fix index table

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-11-21 18:58:54 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_audio_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_automatic_speech_recognition.py Adding chunking for whisper (all seq2seq actually). Very crude matching algorithm. (#20104) 2022-11-14 22:32:50 +01:00
test_pipelines_common.py Add Audio Spectogram Transformer (#19981) 2022-11-21 18:58:54 +01:00
test_pipelines_conversational.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_depth_estimation.py Add depth estimation pipeline (#18618) 2022-10-12 08:54:20 -04:00
test_pipelines_document_question_answering.py A few CI fixes for DocumentQuestionAnsweringPipeline (#19584) 2022-10-17 15:35:27 +02:00
test_pipelines_feature_extraction.py add return_tensors parameter for feature_extraction 2 (#19707) 2022-10-18 16:29:00 +02:00
test_pipelines_fill_mask.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_image_classification.py Specify TF framework explicitly in more pipeline tests (#19748) 2022-10-19 16:24:03 +02:00
test_pipelines_image_segmentation.py Make ImageSegmentationPipelineTests less flaky (#20147) 2022-11-15 09:14:55 +01:00
test_pipelines_image_to_text.py fix image2test args forwarding (#19648) 2022-10-24 09:49:24 -04:00
test_pipelines_object_detection.py Adding support for LayoutLMvX variants for object-detection. (#20143) 2022-11-10 11:33:38 +01:00
test_pipelines_question_answering.py Clean up deprecation warnings (#19654) 2022-10-18 13:34:47 -04:00
test_pipelines_summarization.py Add Switch transformers (#19323) 2022-11-15 13:06:45 +01:00
test_pipelines_table_question_answering.py Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
test_pipelines_text_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_text_generation.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_text2text_generation.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_token_classification.py TF port of ESM (#19587) 2022-10-17 14:16:16 +01:00
test_pipelines_translation.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_visual_question_answering.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_zero_shot_image_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_zero_shot_object_detection.py [Proposal] Breaking change zero-shot-object-detection for improved consistency. (#20280) 2022-11-18 15:57:28 +01:00
test_pipelines_zero_shot.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00