From a2fc43af0b1e67b9c4c17636d229fc23e37050f0 Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Tue, 12 Nov 2024 16:41:47 +0100
Subject: [PATCH] name correction
---
yolov5/detect.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/yolov5/detect.py b/yolov5/detect.py
index 719b3fd..33b8239 100755
--- a/yolov5/detect.py
+++ b/yolov5/detect.py
@@ -103,12 +103,12 @@ def run(
date_now = date.today().strftime("%Y%m%d")
if sound:
- name = '_'.join([project_name, 'detect', date_now, weights[0].split('/')[-3], 'conf',
+ name = '_'.join([project_name, 'detect', date_now, weights[0].split('/')[-1], 'conf',
str(conf_thres).replace('.','_'), str(rf), str(sampleDur), str(window),
str(int(window*hop)),'low',str(int(low)) if low is not None else 'None',
'high',str(int(high)) if high is not None else 'None',''])
else:
- name = '_'.join([project_name, 'detect', date_now, weights[0].split('/')[-3], 'conf',
+ name = '_'.join([project_name, 'detect', date_now, weights[0].split('/')[-1], 'conf',
str(conf_thres).replace('.','_'),''])
print(f'\nYour train results will be saved as {name}\n')
--
GitLab