From 03f4713eed065d519dd53cca8768a32e28e87b68 Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Sun, 21 Jul 2024 22:09:17 +0200
Subject: [PATCH] Update file train.py
---
yolov5/train.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/yolov5/train.py b/yolov5/train.py
index ee3be00..304f95b 100755
--- a/yolov5/train.py
+++ b/yolov5/train.py
@@ -469,7 +469,7 @@ def parse_opt(known=False):
parser.add_argument('--save-period', type=int, default=-1, help='Save checkpoint every x epochs (disabled if < 1)')
parser.add_argument('--seed', type=int, default=0, help='Global training seed')
parser.add_argument('--local_rank', type=int, default=-1, help='Automatic DDP Multi-GPU argument, do not modify')
- parser.add_argument('--sr', type=str, default='', help='If sound, give the samplerate of the spectrogram')
+ parser.add_argument('--rf', type=str, default='', help='If sound, give the samplerate of the spectrogram')
parser.add_argument('--duration', type=str, default='', help='If sound, give the duration of the spectrogram')
# Logger arguments
@@ -514,7 +514,7 @@ def main(opt, callbacks=Callbacks()):
opt.name = Path(opt.cfg).stem # use model.yaml as name
project_name = input('Please enter the name of your project : ')
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.sr, opt.duration, 'YOLOV5',])
+ 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}')
opt.save_dir = str(increment_path(Path(opt.project) / folder_name, exist_ok=opt.exist_ok))
--
GitLab