diff --git a/utils/datasets.py b/utils/datasets.py
index b97ef7b4..c3fdaea4 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -576,19 +576,19 @@ def load_mosaic(self, index):
     # Concat/clip labels
     if len(labels4):
         labels4 = np.concatenate(labels4, 0)
-        # np.clip(labels4[:, 1:], 0, 2 * s, out=labels4[:, 1:])  # use before random_affine
-        np.clip(labels4[:, 1:], s / 2, 1.5 * s, out=labels4[:, 1:])
-        labels4[:, 1:] -= s / 2
+        np.clip(labels4[:, 1:], 0, 2 * s, out=labels4[:, 1:])  # use before random_affine
+        # np.clip(labels4[:, 1:], s / 2, 1.5 * s, out=labels4[:, 1:])
+        # labels4[:, 1:] -= s / 2
 
-    img4 = img4[s // 2: int(s * 1.5), s // 2:int(s * 1.5)]
+    # img4 = img4[s // 2: int(s * 1.5), s // 2:int(s * 1.5)]
 
-    # # Augment
-    # img4, labels4 = random_affine(img4, labels4,
-    #                               degrees=self.hyp['degrees'],
-    #                               translate=self.hyp['translate'],
-    #                               scale=self.hyp['scale'],
-    #                               shear=self.hyp['shear'],
-    #                               border=-s // 2)  # border to remove
+    # Augment
+    img4, labels4 = random_affine(img4, labels4,
+                                  degrees=self.hyp['degrees'],
+                                  translate=self.hyp['translate'],
+                                  scale=self.hyp['scale'],
+                                  shear=self.hyp['shear'],
+                                  border=-s // 2)  # border to remove
 
     return img4, labels4