diff --git a/get_train_annot_YOLO.py b/get_train_annot_YOLO.py index f3efb59d22ffc7e452d331a730b11f48bc5fa83c..e3d3a2df6388b66be2582b11da884db477ed69fe 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))))