42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
# -*- 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)
|