transformers/tests/models/vitmatte
Henrik Matthiesen a847d4aa6b
Fast image processor for VitMatte added and bug in slow version fixed (#37616)
* added fast image processor for VitMatte including updated and new tests, fixed a bug in the slow image processor that processed images incorrectly for input format ChannelDimension.FIRST in which case the trimaps were not added in the correct dimension, this bug was also reflected in the tests through incorretly shaped trimaps being passed

* final edits for fast vitmatte image processor and tests

* final edits for fast vitmatte image processor and tests

---------

Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com>
2025-04-28 14:51:50 -04:00
..
__init__.py Add ViTMatte (#25843) 2023-09-19 10:56:10 -03:00
test_image_processing_vitmatte.py Fast image processor for VitMatte added and bug in slow version fixed (#37616) 2025-04-28 14:51:50 -04:00
test_modeling_vitmatte.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00