Skip to content

Commit 1db18d6

Browse files
committed
cpu boxes fix
1 parent 7e5db1f commit 1db18d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vision/workflow-test/detector_rewrite_live.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def main():
9696
tmp.unique_object_id = sl.generate_unique_id()
9797
tmp.probability = box.conf
9898
tmp.label = int(box.cls)
99-
tmp.bounding_box_2d = np.array(box.xyxy)
99+
tmp.bounding_box_2d = box.xyxy.cpu().numpy()
100100
tmp.is_grounded = True
101101
objects_in.append(tmp)
102102

0 commit comments

Comments
 (0)