pipeline.py 等更新

2025-01-21 18:22:56 +08:00
parent bfe7bc0fd5
commit 64248b1557
15 changed files with 106 additions and 77 deletions
--- a/track_reid.py
+++ b/track_reid.py
@ -216,7 +216,7 @@ def yolo_resnet_tracker(
                # Rescale boxes from img_size to im0 size
                det[:, :4] = scale_boxes(im.shape[2:], det[:, :4], im0.shape).round()
                
-                # det = det.cpu().numpy()                
+                det = det.cpu().numpy()                
                ## ================================================================ writed by WQG            
                '''tracks: [x1, y1, x2, y2, track_id, score, cls, frame_index, box_index]
                            0   1   2   3      4        5     6       7           8