modify pipeline.py

This commit is contained in:
王庆刚
2025-01-13 17:35:15 +08:00
parent 3d13b0d9c5
commit a50f777839
12 changed files with 322 additions and 211 deletions

View File

@ -26,12 +26,56 @@
4. 整体流程仿真
pipeline.py
SourceType: "image", "video", yolo+resent+tracker模块输入数据类型
需要指定是个变量:
(1) evtdir: 保存各个事件的文件夹
(2) SourceType: "image", "video", yolo+resent+tracker模块输入数据类型
(3) savepath: 结果数据保存位置
(4) weights: YoloV5 模型权重
savepath
(1) Yolos_Tracking
eventname文件夹保存yolo+resent+tracker模块输出包括
(1) images 或 {videoname}
(2) subimgs
(3) trajectory
(4) front_y2.png
(5) trajectory.png
(2) ShoppingDict_pkfile
保存 ShoppingDict 字典的 pickle 文件,可用于构造 ShoppingEvent 对象。
(3) contrast
保存1:1、1:SN、1:n数据
(4) error_events.txt
事件字典结构:
ShoppingDict = {"eventPath": eventpath,
"eventName": evtname,
"barcode": barcode,
"eventType": '', # "input", "output", "other"
"frontCamera": {},
"backCamera": {},
"one2n": []
}
保存为 pickle 文件,该文件可用于 ShoppingEvent 对象构建。
相机字典:
CameraEvent = {"cameraType": '', # "front", "back"
"videoPath": '',
"imagePaths": [],
"yoloResnetTracker": [],
"tracking": [],
}
yoloResnetTracker列表元素 "frameDict" 为有关帧数据的字典结构
frameDict = {"path": path,
"fid": int(frameId),
"bboxes": det,
"tboxes": tracks,
"imgs": imgdict,
"feats": featdict}