tests/models/seggpt/test_image_processing_seggpt.py · 6d4cabda2614d86357092585b416c4d08be73382 · chenpangpang / transformers

[SegGPT] Fix seggpt image processor (#29550) · 6d4cabda

Eduardo Pacheco authored Apr 26, 2024

* 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

6d4cabda

test_image_processing_seggpt.py 13.1 KB

Replace test_image_processing_seggpt.py