ObjectDetectionTask: increase test coverage for torchvision 0.14+ (#1739

) * fix test coverage in trainers validation step * add tests for multilabelclassificationtask * Minimal bug fix for now * Simpler random code --------- Co-authored-by: Adam J. Stewart <[email protected]>
microsoft · Mar 2, 2024 · d147be1 · d147be1
1 parent bae6988
commit d147be1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/trainers/test_detection.py b/tests/trainers/test_detection.py
@@ -45,9 +45,12 @@ def forward(self, images: Any, targets: Any = None) -> Any:
         else:  # eval mode
             output = []
             for i in range(batch_size):
+                boxes = torch.rand(10, 4)
+                # xmax, ymax must be larger than xmin, ymin
+                boxes[:, 2:] += 1
                 output.append(
                     {
-                        "boxes": torch.rand(10, 4),
+                        "boxes": boxes,
                         "labels": torch.randint(0, 2, (10,)),
                         "scores": torch.rand(10),
                     }