From d859957c66295dab8174a36e43a7c62c79a29855 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 27 Dec 2019 12:34:29 -0800 Subject: [PATCH] updates --- utils/datasets.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index c7d6a017..b88b769c 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -584,10 +584,10 @@ def load_mosaic(self, index): # Augment # img4 = img4[s // 2: int(s * 1.5), s // 2:int(s * 1.5)] # center crop (WARNING, requires box pruning) img4, labels4 = random_affine(img4, labels4, - degrees=self.hyp['degrees'] * 1, - translate=self.hyp['translate'] * 1, - scale=self.hyp['scale'] * 1, - shear=self.hyp['shear'] * 1, + degrees=self.hyp['degrees'] * 0, + translate=self.hyp['translate'] * 0, + scale=self.hyp['scale'] * 0, + shear=self.hyp['shear'] * 0, border=-s // 2) # border to remove return img4, labels4