Fix Bboxes numpy.reshape (#17301)

2024-11-01 21:08:30 +08:00 · 2024-11-01 21:08:30 +08:00 · 591fdbd8b1
commit 591fdbd8b1
parent 3a4b65c347
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/utils/instance.py
+++ b/ultralytics/utils/instance.py
@ -176,7 +176,7 @@ class Bboxes:
            length as the number of bounding boxes.
        """
        if isinstance(index, int):
-            return Bboxes(self.bboxes[index].view(1, -1))
+            return Bboxes(self.bboxes[index].reshape(1, -1))
        b = self.bboxes[index]
        assert b.ndim == 2, f"Indexing on Bboxes with {index} failed to return a matrix!"
        return Bboxes(b)