From abdd8f57371f2710a3517a3158d54f909aa4c2ce Mon Sep 17 00:00:00 2001 From: jiajie555 Date: Sun, 27 Apr 2025 16:48:47 +0800 Subject: [PATCH] '' --- pipeline.py | 6 +++++- pipeline_process_addWrite.py | 7 ++++--- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/pipeline.py b/pipeline.py index afdc6f0..4c05943 100644 --- a/pipeline.py +++ b/pipeline.py @@ -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 save_path = r"D:\work\result_pipeline", kk=1, @@ -364,6 +365,9 @@ def execute_pipeline(evtdir = r"D:\datasets\ym\后台数据\unzip", f.write(line + '\n') 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/' savepath = r'/home/wqg/dataset/pipeline/test_result/single_event_V10' diff --git a/pipeline_process_addWrite.py b/pipeline_process_addWrite.py index 323172d..d5cf2ad 100644 --- a/pipeline_process_addWrite.py +++ b/pipeline_process_addWrite.py @@ -538,15 +538,16 @@ if __name__ == "__main__": datapath = '/home/yujia/yj/gpu_code/callback_data_test_0417/' savepath = '/home/yujia/yj/gpu_code/result_0417_v10/' max_col = 12 ##excel表格列索引从0开始，从这列开始写入代码解析内容 - track_txt = '轨迹数为空.txt' + # track_txt = '轨迹数为空.txt' + track_txt = '' ###第一次跑pipeline时，track_txt为空 execute_pipeline(evtdir=datapath, DataType = "raw", # raw, pkl kk=1, source_type = "video", # video, image, save_path = savepath, - yolo_ver = "V5", # V10, V5 ##20250401之前使用V5 ressnet使用resv10 - weight_yolo_v5 = '/home/yujia/yj/gpu_code/ckpts/best_cls10_0906.pt' , + yolo_ver = "V10", # V10, V5 ##20250401之前使用V5 ressnet使用resv10 + 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', saveimages = False, max_col = max_col,