From bdb2f0aa7d37bcb4ca50b1f4a9f54ec8f220d296 Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephanechvn@gmail.com>
Date: Tue, 18 Apr 2023 14:54:34 +0200
Subject: [PATCH] Update get_train_annot_YOLO.py
---
get_train_annot_YOLO.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/get_train_annot_YOLO.py b/get_train_annot_YOLO.py
index f3efb59..e3d3a2d 100644
--- a/get_train_annot_YOLO.py
+++ b/get_train_annot_YOLO.py
@@ -167,12 +167,12 @@ def process(x):
for l in range(len(fin)):
try :
- plt.gca().add_patch(Rectangle(((fin.x.iloc[l]*len(time))-(0.5*fin.w.iloc[l]*len(time)), ((fin.y.iloc[l]*len(freq))+0.5*fin.height.iloc[l]*len(freq))), fin.width.iloc[l]*len(time), fin.height.iloc[l]*len(freq), linewidth=3, edgecolor=color[int(fin.id.iloc[l])], facecolor='none'))
+ plt.gca().add_patch(Rectangle(((fin.x.iloc[l]*len(time))-(0.5*fin.width.iloc[l]*len(time)), ((fin.y.iloc[l]*len(freq))+0.5*fin.height.iloc[l]*len(freq))), fin.width.iloc[l]*len(time), fin.height.iloc[l]*len(freq), linewidth=3, edgecolor=color[int(fin.id.iloc[l])], facecolor='none'))
except IndexError:
ipdb.set_trace()
try:
- plt.savefig(os.path.join(directrory, str('images_annotes_'+str(today.day)+'_'+str(today.month)),str(name+'.jpg')))
+ plt.savefig(os.path.join(directory, str('images_annotes_'+str(today.day)+'_'+str(today.month)),str(name+'.jpg')))
except Exception:
os.mkdir(os.path.join(directory,str('images_annotes_'+str(today.day)+'_'+str(today.month))))
--
GitLab