transformers/tests/models/sam
Rosie Wood 73c88012b7
Add segmentation map processing to SAM Image Processor (#27463)
* add segmentation map processing to sam image processor

* fixup

* add tests

* reshaped_input_size is shape before padding

* update tests for size/shape outputs

* fixup

* add code snippet to docs

* Update docs/source/en/model_doc/sam.md

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Add missing backticks

* add `segmentation_maps` as arg for SamProcessor.__call__()

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-01-08 16:40:36 +00:00
..
__init__.py Add Segment Anything Model (SAM) (#22654) 2023-04-19 21:01:49 +02:00
test_modeling_sam.py Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
test_modeling_tf_sam.py [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
test_processor_sam.py Add segmentation map processing to SAM Image Processor (#27463) 2024-01-08 16:40:36 +00:00