# -*- coding: utf-8 -*- """ Created on Fri Mar 28 11:35:28 2025 @author: ym """ from pipeline import execute_pipeline def execute(datapath, savepath_v5, savepath_v10): execute_pipeline(evtdir = datapath, DataType = "raw", # raw, pkl kk=None, source_type = "video", # video, image, save_path = savepath_v5, yolo_ver = "V5", # V10, V5 weight_yolo_v5 = r'./ckpts/best_cls10_0906.pt' , weight_yolo_v10 = r'./ckpts/best_v10s_width0375_1205.pt', saveimages = False ) execute_pipeline(evtdir = datapath, DataType = "raw", # raw, pkl kk=None, source_type = "video", # video, image, save_path = savepath_v10, yolo_ver = "V10", # V10, V5 weight_yolo_v5 = r'./ckpts/best_cls10_0906.pt' , weight_yolo_v10 = r'./ckpts/best_v10s_width0375_1205.pt', saveimages = False ) datapath = r'/home/wqg/dataset/test_dataset/base_dataset/single_event/source/' savepath_v5 = r'/home/wqg/dataset/pipeline/contrast/single_event_V5' savepath_v10 = r'/home/wqg/dataset/pipeline/contrast/single_event_V10' execute(datapath, savepath_v5, savepath_v10) datapath = r'/home/wqg/dataset/test_performence_dataset/' savepath_v5 = r'/home/wqg/dataset/pipeline/contrast/performence_V5' savepath_v10 = r'/home/wqg/dataset/pipeline/contrast/performence_V10' execute(datapath, savepath_v5, savepath_v10)