From 11710a5fbee3d21d2be75f1c080104ef0c7b948a Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Tue, 8 Apr 2025 10:32:09 +0200
Subject: [PATCH] Update 2 files
- /yolov5/detect.py
- /yolov5/train.py
---
yolov5/detect.py | 2 +-
yolov5/train.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/yolov5/detect.py b/yolov5/detect.py
index 34beba0..cd69d9c 100755
--- a/yolov5/detect.py
+++ b/yolov5/detect.py
@@ -98,7 +98,7 @@ def run(
screenshot = source.lower().startswith('screen')
if is_url and is_file:
source = check_file(source) # download
- if not args.name:
+ if not name:
project_name = input('Please enter the name of your project : ')
else:
project_name = args.name
diff --git a/yolov5/train.py b/yolov5/train.py
index e18d4ea..c1e109f 100755
--- a/yolov5/train.py
+++ b/yolov5/train.py
@@ -512,7 +512,7 @@ def main(opt, callbacks=Callbacks()):
opt.exist_ok, opt.resume = opt.resume, False # pass resume to exist_ok and disable resume
if opt.name == 'cfg':
opt.name = Path(opt.cfg).stem # use model.yaml as name
- if not args.name:
+ if not opt.name:
project_name = input('Please enter the name of your project : ')
else:
project_name = args.name
--
GitLab