Update 2 files

- /yolov5/train.py
- /yolov5/detect.py

yolov5/detect.py

@@ -101,7 +101,7 @@ def run(
     if not name:
         project_name = input('Please enter the name of your project : ')
     else:
-        project_name = args.name
+        project_name = name
     date_now = date.today().strftime("%Y%m%d")
 
     if sound:

yolov5/train.py

@@ -515,7 +515,7 @@ def main(opt, callbacks=Callbacks()):
         if not opt.name:
             project_name = input('Please enter the name of your project : ')
         else:
-            project_name = args.name
+            project_name = opt.name
         date_now = date.today().strftime("%Y%m%d")
         folder_name = '_'.join([project_name, date_now, opt.weights[-4:-3], str(opt.imgsz), opt.optimizer, opt.rf, opt.duration, 'YOLOV5',])
         print(f'Your train results will be saved in {folder_name}')