Simpler random code

microsoft · Nov 25, 2023 · 0f5f89e · 0f5f89e
1 parent d4e7579
commit 0f5f89e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/trainers/test_detection.py b/tests/trainers/test_detection.py
@@ -45,9 +45,9 @@ def forward(self, images: Any, targets: Any = None) -> Any:
         else:  # eval mode
             output = []
             for i in range(batch_size):
-                boxes = torch.zeros(10, 4, dtype=torch.float)
-                # Create xmax, ymax larger than 0.0
-                boxes[:, 2:4] = torch.FloatTensor(10, 2).uniform_(0.1, 0.9)
+                boxes = torch.rand(10, 4)
+                # xmax, ymax must be larger than xmin, ymin
+                boxes[:, 2:] += 1
                 output.append(
                     {
                         "boxes": boxes,