tests/models/detr/test_image_processing_detr.py · bd4b83e1ba52904e4917ac41dbbd10cd41803d0b · chenpangpang / transformers

[`DETR`] Update the processing to adapt masks & bboxes to reflect padding (#28363) · bd4b83e1

amyeroberts authored Feb 13, 2024

* Update the processing so bbox coords are adjusted for padding

* Just pad masks

* Tidy up, add tests

* Better tests

* Fix yolos and mark as slow for pycocotols

* Fix yolos - return_tensors

* Clarify padding and normalization behaviour

bd4b83e1

test_image_processing_detr.py 24.5 KB

Replace test_image_processing_detr.py