yujia/ieemoo-ai-detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

yolo修改为v5模型

Browse Source
This commit is contained in:
jiajie555
2025-04-18 17:00:02 +08:00
parent 7c87e776e1
commit e6803a29dd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pipeline_process_addWrite.py
View File

@ -545,7 +545,7 @@ if __name__ == "__main__":
kk=1, kk=1,
source_type = "video", # video, image, source_type = "video", # video, image,
save_path = savepath, save_path = savepath,
yolo_ver = "V10", # V10, V5 yolo_ver = "V5", # V10, V5 ##20250401之前使用V5 ressnet使用resv10
weight_yolo_v5 = '/home/yujia/yj/gpu_code/ckpts/best_cls10_0906.pt' , 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', weight_yolo_v10 = '/home/yujia/yj/gpu_code/ckpts/best_v10s_width0375_1205.pt',
saveimages = False, saveimages = False,