# -*- coding: utf-8 -*-
"""
Created on Fri Mar 28 11:35:28 2025

@author: ym
"""

from pipeline_01 import execute_pipeline


execute_pipeline(evtdir = r"D:\datasets\ym\后台数据\unzip", 
                 DataType = "raw",   # raw, pkl
                 kk=1,
                 source_type = "video",   # video, image,       
                 save_path = r"D:\work\result_pipeline_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 = r"D:\datasets\ym\后台数据\unzip", 
                 DataType = "raw",   # raw, pkl
                 kk=1,
                 source_type = "video",   # video, image,    
                 save_path = r"D:\work\result_pipeline_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
                )