transformers/tests/models/vitmatte
NielsRogge 7d6354e047
Add ViTMatte (#25843)
* First draft

* Simplify image processor

* Fix rebase

* Address comments

* Address more comments

* Address more comments

* Address more comments

* Address more comments

* Improve pad_image

* Add tests

* Update integration test

* Fix image processor tests

* Fix model tests

* Convert checkpoints

* Fix doc tests

* Remove file

* Apply suggestions

* Address comments

* Fix typing hint

* Add batch_norm_eps

* Address comments

* Fix style
2023-09-19 10:56:10 -03:00
..
__init__.py Add ViTMatte (#25843) 2023-09-19 10:56:10 -03:00
test_image_processing_vitmatte.py Add ViTMatte (#25843) 2023-09-19 10:56:10 -03:00
test_modeling_vitmatte.py Add ViTMatte (#25843) 2023-09-19 10:56:10 -03:00