diff --git a/get_train_annot_YOLO.py b/get_train_annot_YOLO.py index 0980f0a1236750d97f9cafad488071c07580a8b6..19d4b1c2fac73c020f8df0173a11cdc75d728793 100644 --- a/get_train_annot_YOLO.py +++ b/get_train_annot_YOLO.py @@ -185,7 +185,7 @@ def process(x): H,W = im.shape[0], im.shape[1] for l in range(len(fin)): - x, y, w, h = fin.x.iloc[l]*W , fin.y.iloc[l]*H , fin.w.iloc[l]*W , fin.h.iloc[l]*H + x, y, w, h = fin.x.iloc[l]*W , fin.y.iloc[l]*H , fin.width.iloc[l]*W , fin.height.iloc[l]*H shape1 = (int(x-(0.5*w)), int(y+(0.5*h))) shape2 = (int(x+(0.5*w)), int(y+(0.5*h))) @@ -210,6 +210,7 @@ def process(x): cv2.putText(im, label, text_shape, cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255,255,255), 1) plt.imshow(im) + plt.subplots_adjust(top=1, bottom=0, left=0, right=1) try: plt.savefig(os.path.join(directory, str('images_annotes_'+str(today.day)+'_'+str(today.month)),str(name+'.jpg')))