From 32a52dfb02873b12bca6da71cf18bfe172bfa7d1 Mon Sep 17 00:00:00 2001 From: glenn-jocher Date: Fri, 5 Jul 2019 12:33:37 +0200 Subject: [PATCH] GIoU to default --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 33d6bb9f..48dc60a0 100644 --- a/train.py +++ b/train.py @@ -284,7 +284,7 @@ def print_mutation(hyp, results): if opt.cloud_evolve: os.system('gsutil cp gs://yolov4/evolve.txt .') # download evolve.txt - with open('evolve.txt', 'a') as f: # append result to evolve.txt + with open('evolve.txt', 'a') as f: # append result f.write(c + b + '\n') os.system('gsutil cp evolve.txt gs://yolov4') # upload evolve.txt else: @@ -332,7 +332,7 @@ if __name__ == '__main__': print_mutation(hyp, results) # Write mutation results for _ in range(gen): - # Get best hyperparamters + # Get best hyperparameters x = np.loadtxt('evolve.txt', ndmin=2) fitness = x[:, 2] * 0.9 + x[:, 3] * 0.1 # fitness as weighted combination of mAP and F1 x = x[fitness.argmax()] # select best fitness hyps