Fix `Bboxes` numpy.reshape (#17301)

pull/17308/head^2
Laughing 4 weeks ago committed by GitHub
parent 3a4b65c347
commit 591fdbd8b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      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)

Loading…
Cancel
Save