20250313

pipeline.py

@@ -166,12 +166,15 @@ def pipeline(
         
         '''================= 4. tracking ================='''
         '''(1) 生成用于 tracking 模块的 boxes、feats'''  
+        bboxes = np.empty((0, 6), dtype=np.float64)
         trackerboxes = np.empty((0, 9), dtype=np.float64)
         trackefeats = {}
         for frameDict in yrtOut:            
             tboxes = frameDict["tboxes"]
             ffeats = frameDict["feats"]
             
+            boxes = frameDict["bboxes"]
+            bboxes = np.concatenate((bboxes, np.array(boxes)), axis=0)
             trackerboxes = np.concatenate((trackerboxes, np.array(tboxes)), axis=0)
             for i in range(len(tboxes)):
                 fid, bid = int(tboxes[i, 7]), int(tboxes[i, 8])
@@ -266,7 +269,7 @@ def main():
     函数：pipeline()，遍历事件文件夹，选择类型 image 或 video,
     '''     
     parmDict = {}
-    evtdir = r"D:\全实时\202502"
+    evtdir = r"\\192.168.1.28\share\测试视频数据以及日志\全实时测试\V12\2025-3-3"
     parmDict["SourceType"] = "video"   # video, image
     parmDict["savepath"] = r"D:\全实时\202502\result"
     parmDict["weights"] = r'D:\DetectTracking\ckpts\best_cls10_0906.pt' 
@@ -275,10 +278,9 @@ def main():
     k, errEvents = 0, []
     for item in evtdir.iterdir():
         if item.is_dir():            
-            item = evtdir/Path("20250228-160049-188_6921168558018_6921168558018")
+            item = evtdir/Path("20250303-103058-074_6914973604223_6914973604223")
             parmDict["eventpath"] = item
             # pipeline(**parmDict)
-            
             try:
                 pipeline(**parmDict)
             except Exception as e: