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/conditional_detr/test_image_processing_conditional_detr.py
View File

@@ -368,7 +368,6 @@ class ConditionalDetrImageProcessingTest(AnnotationFormatTestMixin, ImageProcess
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
# Copied from tests.models.detr.test_image_processing_detr.DetrImageProcessingTest.test_batched_coco_panoptic_annotations with Detr->ConditionalDetr
def test_batched_coco_panoptic_annotations(self):
# prepare image, target and masks_path