From bab855507a1e99e6c943292eab7f2ffd2de059b2 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 17 Jan 2020 18:05:28 -0800 Subject: [PATCH] updates --- train.py | 2 +- utils/datasets.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/train.py b/train.py index 859368af..195e4a6c 100644 --- a/train.py +++ b/train.py @@ -221,7 +221,7 @@ def train(): # Prebias if prebias: - if epoch < 20: # prebias + if epoch < 1: # prebias ps = 0.1, 0.9 # prebias settings (lr=0.1, momentum=0.9) else: # normal training ps = hyp['lr0'], hyp['momentum'] # normal training settings diff --git a/utils/datasets.py b/utils/datasets.py index 3f8a8304..f83a17be 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -593,10 +593,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'] * 0, - translate=self.hyp['translate'] * 0, - scale=self.hyp['scale'] * 0, - shear=self.hyp['shear'] * 0, + degrees=self.hyp['degrees'] * 1, + translate=self.hyp['translate'] * 1, + scale=self.hyp['scale'] * 1, + shear=self.hyp['shear'] * 1, border=-s // 2) # border to remove return img4, labels4