From 9835fcc64e46deef942f29192a2bb3b1ed67797b Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Sat, 15 Apr 2023 13:12:32 +0300 Subject: [PATCH] Perspective Aug #0016 --- utils/dataloaders.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 048b3078252d..a0c2585a7388 100644 --- 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)