transformers/tests/models/seggpt
Eduardo Pacheco 6d4cabda26
[SegGPT] Fix seggpt image processor (#29550)
* Fixed SegGptImageProcessor to handle 2D and 3D prompt mask inputs

* Added new test to check prompt mask equivalence

* New proposal

* Better proposal

* Removed unnecessary method

* Updated seggpt docs

* Introduced do_convert_rgb

* nits
2024-04-26 19:40:12 +01:00
..
__init__.py Adding SegGPT (#27735) 2024-02-26 18:17:19 +00:00
test_image_processing_seggpt.py [SegGPT] Fix seggpt image processor (#29550) 2024-04-26 19:40:12 +01:00
test_modeling_seggpt.py [SegGPT] Fix seggpt image processor (#29550) 2024-04-26 19:40:12 +01:00