transformers/tests
amyeroberts de6d19ea92
Add segmentation + object detection image processors (#20160)
* Add transforms for object detection

* DETR models + Yolos

* Scrappy additions

* Maskformer image processor

* Fix up; MaskFormer tests

* Update owlvit processor

* Add to docs

* OwlViT tests

* Update pad logic

* Remove changes to transforms

* Import fn directly

* Update to include pad transformation

* Remove uninstended changes

* Add new owlvit post processing function

* Tidy up

* Fix copies

* Fix some copies

* Include device fix

* Fix scipy imports

* Update _pad_image

* Update padding functionality

* Fix bug

* Properly handle ignore index

* Fix up

* Remove defaults to None in docstrings

* Fix docstrings & docs

* Fix sizes bug

* Resolve conflicts in init

* Cast to float after resizing

* Tidy & add size if missing

* Allow kwards when processing for owlvit

* Update test values
2022-11-30 10:24:03 +00:00
..
benchmark
deepspeed Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
extended
fixtures
generation Generate: model_kwargs can also be an input to prepare_inputs_for_generation (#20353) 2022-11-21 16:20:27 +00:00
mixed_int8 [BNB] Throw ValueError when trying to cast or assign (#20409) 2022-11-23 15:51:50 +01:00
models Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
onnx Use tiny models for ONNX tests - text modality (#20333) 2022-11-22 17:11:17 +01:00
optimization
pipelines Add Audio Spectogram Transformer (#19981) 2022-11-21 18:58:54 +01:00
repo_utils Repo utils test (#19696) 2022-10-18 13:47:36 -04:00
sagemaker
tokenization
trainer [AnyPrecisionAdamW] test fix (#20454) 2022-11-25 09:02:10 -08:00
utils Include image processor in add-new-model-like (#20439) 2022-11-28 16:46:02 +00:00
__init__.py
test_configuration_common.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_common.py
test_image_transforms.py [Image Transformers] to_pil fix float edge cases (#20406) 2022-11-23 13:47:59 +01:00
test_modeling_common.py [AutoBackbone] Improve API (#20407) 2022-11-28 17:20:24 +01:00
test_modeling_flax_common.py Allow flax subfolder (#19902) 2022-10-26 18:33:23 +02:00
test_modeling_tf_common.py More TF int dtype fixes (#20384) 2022-11-28 13:24:44 +00:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py 🚨 🚨 🚨 Fix Issue 15003: SentencePiece Tokenizers Not Adding Special Tokens in convert_tokens_to_string (#15775) 2022-11-02 15:45:38 -04:00