Skip to content

Commit 37d9ee2

Browse files
committed
Change return object from tensor object to list object
1 parent 88d9224 commit 37d9ee2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

homework/person_detector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ def person_detector(img: cv2.Mat):
99

1010
# Only detect people (class id: 0) within the image
1111
results = model.predict(img, classes=[0])
12-
return results[0].boxes.xywh
12+
return results[0].boxes.xywh.tolist()
1313

1414

1515
def main(fn):

0 commit comments

Comments
 (0)