lichen/detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

20241218

Browse Source
This commit is contained in:
王庆刚
2024-12-18 17:35:24 +08:00
parent 39f94c7bd4
commit dac3b3f2b6
8 changed files with 862 additions and 787 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
contrast/event_test.py
View File

@ -7,7 +7,7 @@ Created on Mon Dec 16 18:56:18 2024
import os
import cv2
from utils.event import ShoppingEvent
from utils.event import ShoppingEvent, save_data
def main():
evtpaths = r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\images"
@ -29,39 +29,14 @@ def main():
events = list(set(events))
'''定义当前事件存储地址及生成相应文件件'''
resultPath = r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\result"
resultPath = r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\result\single_event"
for evtpath in events:
evtname = os.path.basename(evtpath)
event = ShoppingEvent(evtpath)
save_data(event, resultPath)
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)
print(event.evtname)
## 保存序列图像和轨迹子图
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}")