modify the event pickle file

pipeline.py

@@ -114,6 +114,7 @@ def pipeline(
                     "backCamera": {},
                     "one2n": []        # 
                     }
+    yrtDict = {}
     
     
     procpath = Path(eventpath).joinpath('process.data')
@@ -158,11 +159,23 @@ def pipeline(
         
         '''================= 3. Yolo + Resnet + Tracker ================='''
         optdict["source"] = vpath
-        optdict["save_dir"] = savepath_pipeline_imgs        
+        optdict["save_dir"] = savepath_pipeline_imgs   
+        optdict["is_save_img"] = False
+        optdict["is_save_video"] = True
+
         
         yrtOut = yolo_resnet_tracker(**optdict)
         
-        CameraEvent["yoloResnetTracker"] = yrtOut
+        yrtOut_save = []
+        for frdict in yrtOut:
+            fr_dict = {}
+            for k, v in frdict.items():
+                if k != "imgs":
+                    fr_dict[k]=v
+            yrtOut_save.append(fr_dict) 
+        CameraEvent["yoloResnetTracker"] = yrtOut_save
+        
+        # CameraEvent["yoloResnetTracker"] = yrtOut
         
         '''================= 4. tracking ================='''
         '''(1) 生成用于 tracking 模块的 boxes、feats'''  
@@ -189,6 +202,8 @@ def pipeline(
             
             CameraEvent["tracking"] = vts
             ShoppingDict["backCamera"] = CameraEvent
+            
+            yrtDict["backyrt"] = yrtOut
         
         '''(2) tracking, 前摄'''    
         if  CameraEvent["cameraType"] == "front":
@@ -198,6 +213,8 @@ def pipeline(
             
             CameraEvent["tracking"] = vts
             ShoppingDict["frontCamera"] = CameraEvent
+            
+            yrtDict["frontyrt"] = yrtOut
     
     '''========================== 保存模块 ================================='''
     '''(1) 保存 ShoppingDict 事件'''
@@ -208,10 +225,14 @@ def pipeline(
     for CamerType, vts in event_tracks:
         if len(vts.tracks)==0: continue
         if CamerType == 'front':
-            yolos = ShoppingDict["frontCamera"]["yoloResnetTracker"]
+            # yolos = ShoppingDict["frontCamera"]["yoloResnetTracker"]
+            
+            yolos = yrtDict["frontyrt"]
             ctype = 1
         if CamerType == 'back':
-            yolos = ShoppingDict["backCamera"]["yoloResnetTracker"]
+            # yolos = ShoppingDict["backCamera"]["yoloResnetTracker"]
+            
+            yolos = yrtDict["backyrt"]
             ctype = 0
             
         imgdict, featdict, simidict = {}, {}, {}
@@ -269,10 +290,10 @@ def main():
     函数：pipeline()，遍历事件文件夹，选择类型 image 或 video,
     '''     
     parmDict = {}
-    evtdir = r"D:\works\后台回传数据"
+    evtdir = r"../dataset/backend_20250310"
     parmDict["SourceType"] = "video"   # video, image
-    parmDict["savepath"] = r"D:\works\results"
-    parmDict["weights"] = r'D:\DetectTracking\ckpts\best_cls10_0906.pt' 
+    parmDict["savepath"] = r"../dataset/run"
+    parmDict["weights"] = r'./ckpts/best_cls10_0906.pt' 
 
     evtdir = Path(evtdir) 
     k, errEvents = 0, []
@@ -280,14 +301,14 @@ def main():
         if item.is_dir():            
             # item = evtdir/Path("20250303-103058-074_6914973604223_6914973604223")
             parmDict["eventpath"] = item
-            # pipeline(**parmDict)
-            try:
-                pipeline(**parmDict)
-            except Exception as e:
-                errEvents.append(str(item))
-        # k+=1
-        # if k==1:
-        #     break
+            pipeline(**parmDict)
+            # try:
+            #     pipeline(**parmDict)
+            # except Exception as e:
+            #     errEvents.append(str(item))
+        k+=1
+        if k==1:
+            break
             
     errfile = os.path.join(parmDict["savepath"], f'error_events.txt')
     with open(errfile, 'w', encoding='utf-8') as f: