''
This commit is contained in:
@ -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'
|
||||
|
||||
|
@ -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,
|
||||
|
Reference in New Issue
Block a user