diff --git a/pipeline_process_addWrite.py b/pipeline_process_addWrite.py index 55e0a30..323172d 100644 --- a/pipeline_process_addWrite.py +++ b/pipeline_process_addWrite.py @@ -545,7 +545,7 @@ if __name__ == "__main__": kk=1, source_type = "video", # video, image, 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_v10 = '/home/yujia/yj/gpu_code/ckpts/best_v10s_width0375_1205.pt', saveimages = False,