lichen/detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加了单帧入侵判断及yoloV10

Browse Source
This commit is contained in:
18262620154
2025-04-11 17:02:39 +08:00
parent 798c596acc
commit e044c85a04
197 changed files with 1863 additions and 997 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
execute_pipeline.py
View File

@ -5,27 +5,37 @@ Created on Fri Mar 28 11:35:28 2025
@author: ym
"""
from pipeline_01 import execute_pipeline
from pipeline 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
)
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)
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
)
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)