# -*- 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)
