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')))