Perspective Aug ultralytics#16

manole-alexandru · Apr 15, 2023 · 9835fcc · 9835fcc
1 parent 19b1bc7
commit 9835fcc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -695,8 +695,8 @@ def __getitem__(self, index):
                 seg, _, _ = random_perspective(seg, labels, degrees=hyp['degrees'], translate=hyp['translate'], scale=hyp['scale'],
                                                shear=hyp['shear'], perspective=hyp['perspective'], random_parameters=random_parameters)
 
-        cv2.imwrite('runs/image' + str(index) + '.jpg', img)
-        cv2.imwrite('runs/mask' + str(index) + '.png', seg)
+        # cv2.imwrite('runs/image' + str(index) + '.jpg', img)
+        # cv2.imwrite('runs/mask' + str(index) + '.png', seg)
         nl = len(labels)  # number of labels
         if nl:
             labels[:, 1:5] = xyxy2xywhn(labels[:, 1:5], w=img.shape[1], h=img.shape[0], clip=True, eps=1E-3)