HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix Bboxes numpy.reshape (#17301)

Browse source
This commit is contained in:
Laughing 2024-11-01 21:08:30 +08:00 • committed by GitHub
parent 3a4b65c347
commit 591fdbd8b1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/utils/instance.py
View file

@ -176,7 +176,7 @@ class Bboxes:
length as the number of bounding boxes. length as the number of bounding boxes.
""" """
if isinstance(index, int): if isinstance(index, int):
return Bboxes(self.bboxes[index].view(1, -1)) return Bboxes(self.bboxes[index].reshape(1, -1))
b = self.bboxes[index] b = self.bboxes[index]
assert b.ndim == 2, f"Indexing on Bboxes with {index} failed to return a matrix!" assert b.ndim == 2, f"Indexing on Bboxes with {index} failed to return a matrix!"
return Bboxes(b) return Bboxes(b)