SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[YOLOS] Fix - return padded annotations (#29300)

Browse Source 
* Fix yolos processing

* Add back slow marker - protects for pycocotools in slow

* Slow decorator goes above copied from header
This commit is contained in:
amyeroberts
2024-03-01 09:42:13 +00:00
committed by GitHub
parent 0a0a279e99
commit f1b1379f37
9 changed files with 38 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/detr/test_image_processing_detr.py
View File

@@ -426,7 +426,6 @@ class DetrImageProcessingTest(AnnotationFormatTestMixin, ImageProcessingTestMixi
self.assertTrue(torch.allclose(encoding["labels"][0]["boxes"], expected_boxes_0, rtol=1))
self.assertTrue(torch.allclose(encoding["labels"][1]["boxes"], expected_boxes_1, rtol=1))
@slow
def test_batched_coco_panoptic_annotations(self):
# prepare image, target and masks_path
image_0 = Image.open("./tests/fixtures/tests_samples/COCO/000000039769.png")