From 919aff828e4c6a435a747b792403ee7c548629e6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 11 Sep 2019 13:15:16 +0200 Subject: [PATCH] updates --- train.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/train.py b/train.py index 1e30712b..19e248c7 100644 --- a/train.py +++ b/train.py @@ -35,6 +35,11 @@ hyp = {'giou': 1.582, # giou loss gain 'scale': 0.1059, # image scale (+/- gain) 'shear': 0.5768} # image shear (+/- deg) +if os.path.exists('hyp.txt'): # overwrite hyp if hyp.txt is found + x = np.loadtxt('hyp.txt') + for i, k in enumerate(hyp.keys()): + hyp[k] = x[i] + def train(): cfg = opt.cfg