for bakeup

tracking/utils/plotting.py

@@ -286,7 +286,8 @@ def boxing_img(det, img, line_width=3):
     return imgx
 
 def draw_tracking_boxes(imgs, tracks, scale=2):
-    '''tracks: [x1, y1, x2, y2, track_id, score, cls, frame_index, box_index]
+    '''需要确保 imgs 覆盖tracks中的帧ID数
+        tracks: [x1, y1, x2, y2, track_id, score, cls, frame_index, box_index]
                 0   1   2   3      4        5     6       7           8
        关键：
          (1) imgs中的次序和 track 中的 fid 对应
@@ -311,12 +312,13 @@ def draw_tracking_boxes(imgs, tracks, scale=2):
     
     bboxes = array2list(tracks)
     
-    if len(bboxes)!=len(imgs):
-        return []
+    # if len(bboxes)!=len(imgs):
+    #     return False, imgs
     
     subimgs = []
     for i, boxes in enumerate(bboxes):
-        annotator = Annotator(imgs[i].copy())
+        fid = int(boxes[0, 7])
+        annotator = Annotator(imgs[fid-1].copy())
         for *xyxy, tid, conf, cls, fid, bid in boxes:
             label = f'id:{int(tid)}_{int(cls)}_{conf:.2f}'
             
@@ -331,7 +333,7 @@ def draw_tracking_boxes(imgs, tracks, scale=2):
             annotator.box_label(pt2, label, color=color)
         
         img = annotator.result()
-        subimgs.append(img)
+        subimgs.append((fid-1, img))
         
     return subimgs
 

tracking/utils/read_data.py

@@ -110,14 +110,17 @@ def extract_data(datapath):
     
     if len(boxes): 
         trackingboxes.append(np.array(boxes))
-    
+        
     tracking_feat_dict = {}
-    for i, boxes in enumerate(trackingboxes):
-        for box in boxes:
-            tid, fid, bid = int(box[4]), int(box[7]), int(box[8])
-            if f"track_{tid}" not in tracking_feat_dict:
-                tracking_feat_dict[f"track_{tid}"]= {"feats": {}}
-            tracking_feat_dict[f"track_{tid}"]["feats"].update({f"{fid}_{bid}": tracker_feat_dict[f"frame_{fid}"]["feats"][bid]})
+    try:
+        for i, boxes in enumerate(trackingboxes):
+            for box in boxes:
+                tid, fid, bid = int(box[4]), int(box[7]), int(box[8])
+                if f"track_{tid}" not in tracking_feat_dict:
+                    tracking_feat_dict[f"track_{tid}"]= {"feats": {}}
+                tracking_feat_dict[f"track_{tid}"]["feats"].update({f"{fid}_{bid}": tracker_feat_dict[f"frame_{fid}"]["feats"][bid]})
+    except Exception as e:
+        print(f'Path: {datapath}, Error: {e}') 
 
     return bboxes, ffeats, trackerboxes, tracker_feat_dict, trackingboxes, tracking_feat_dict