Change return object from tensor object to list object

krishauser · Feb 4, 2025 · 37d9ee2 · 37d9ee2
1 parent 88d9224
commit 37d9ee2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/homework/person_detector.py b/homework/person_detector.py
@@ -9,7 +9,7 @@ def person_detector(img: cv2.Mat):
 
     # Only detect people (class id: 0) within the image
     results = model.predict(img, classes=[0])
-    return results[0].boxes.xywh
+    return results[0].boxes.xywh.tolist()
 
 
 def main(fn):