20241217

contrast/event_test.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Mon Dec 16 18:56:18 2024
+
+@author: ym
+"""
+import os
+import cv2
+
+from utils.event import ShoppingEvent
+
+def main():
+    evtpaths = r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\images"
+    text1 = "one2n_Error.txt"
+    text2 = "one2SN_Error.txt"
+    events = []
+    text = (text1, text2)
+    for txt in text:
+        txtfile = os.path.join(evtpaths, txt)
+        with open(txtfile, "r") as f:
+            lines = f.readlines()
+            for i, line in enumerate(lines):
+                line = line.strip()
+                if line:
+                    fpath=os.path.join(evtpaths, line)
+                    events.append(fpath)
+                    
+                    
+    events = list(set(events))
+
+    '''定义当前事件存储地址及生成相应文件件'''
+    resultPath = r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\result"    
+    for evtpath in events:
+        evtname = os.path.basename(evtpath)
+        event = ShoppingEvent(evtpath)
+        
+        img_cat = event.draw_tracks()
+        trajpath = os.path.join(resultPath, "trajectory")
+        if not os.path.exists(trajpath):
+            os.makedirs(trajpath)
+        traj_imgpath =  os.path.join(trajpath, evtname+".png")
+        cv2.imwrite(traj_imgpath, img_cat)
+        
+        
+        ## 保存序列图像和轨迹子图
+        subimgpath = os.path.join(resultPath, f"{evtname}", "subimg")
+        imgspath = os.path.join(resultPath, f"{evtname}", "imgs")
+        if not os.path.exists(subimgpath):
+            os.makedirs(subimgpath)
+        if not os.path.exists(imgspath):
+            os.makedirs(imgspath)
+ 
+        
+        subimgpairs = event.save_event_subimg(subimgpath)
+        for subimgName, subimg in subimgpairs:
+            spath = os.path.join(subimgpath, subimgName)
+            cv2.imwrite(spath, subimg)
+
+        imgpairs = event.plot_save_image(imgspath)
+        for imgname, img in imgpairs:
+            spath = os.path.join(imgspath, imgname)
+            cv2.imwrite(spath, img)
+
+        print(f"{evtname}")
+
+    
+
+if __name__ == "__main__": 
+    main()