yujia/ieemoo-ai-detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

''

Browse Source
This commit is contained in:
jiajie555
2025-04-27 16:48:47 +08:00
parent 036a5ab0a7
commit abdd8f5737
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pipeline.py
View File

@ -313,7 +313,8 @@ def pipeline(eventpath,
def execute_pipeline(evtdir = r"D:\datasets\ym\后台数据\unzip", def \
execute_pipeline(evtdir = r"D:\datasets\ym\后台数据\unzip",
DataType = "raw", # raw, pkl DataType = "raw", # raw, pkl
save_path = r"D:\work\result_pipeline", save_path = r"D:\work\result_pipeline",
kk=1, kk=1,
@ -364,6 +365,9 @@ def execute_pipeline(evtdir = r"D:\datasets\ym\后台数据\unzip",
f.write(line + '\n') f.write(line + '\n')
if __name__ == "__main__": if __name__ == "__main__":
# datapath = r'/home/wqg/dataset/test_dataset/base_dataset/single_event/source/'
# savepath = r'/home/wqg/dataset/pipeline/test_result/single_event_V10'
datapath = r'/home/wqg/dataset/test_dataset/base_dataset/single_event/source/' datapath = r'/home/wqg/dataset/test_dataset/base_dataset/single_event/source/'
savepath = r'/home/wqg/dataset/pipeline/test_result/single_event_V10' savepath = r'/home/wqg/dataset/pipeline/test_result/single_event_V10'

7
pipeline_process_addWrite.py
View File

@ -538,15 +538,16 @@ if __name__ == "__main__":
datapath = '/home/yujia/yj/gpu_code/callback_data_test_0417/' datapath = '/home/yujia/yj/gpu_code/callback_data_test_0417/'
savepath = '/home/yujia/yj/gpu_code/result_0417_v10/' savepath = '/home/yujia/yj/gpu_code/result_0417_v10/'
max_col = 12 ##excel表格列索引从0开始，从这列开始写入代码解析内容 max_col = 12 ##excel表格列索引从0开始，从这列开始写入代码解析内容
track_txt = '轨迹数为空.txt' # track_txt = '轨迹数为空.txt'
track_txt = '' ###第一次跑pipeline时，track_txt为空
execute_pipeline(evtdir=datapath, execute_pipeline(evtdir=datapath,
DataType = "raw", # raw, pkl DataType = "raw", # raw, pkl
kk=1, kk=1,
source_type = "video", # video, image, source_type = "video", # video, image,
save_path = savepath, save_path = savepath,
yolo_ver = "V5", # V10, V5 ##20250401之前使用V5 ressnet使用resv10 yolo_ver = "V10", # V10, V5 ##20250401之前使用V5 ressnet使用resv10
weight_yolo_v5 = '/home/yujia/yj/gpu_code/ckpts/best_cls10_0906.pt' , weight_yolo_v5 = '/home/yujia/yj/gpu_code/ckpts/best_cls10_0906.pt',
weight_yolo_v10 = '/home/yujia/yj/gpu_code/ckpts/best_v10s_width0375_1205.pt', weight_yolo_v10 = '/home/yujia/yj/gpu_code/ckpts/best_v10s_width0375_1205.pt',
saveimages = False, saveimages = False,
max_col = max_col, max_col = max_col,