giou updates

ultralytics · May 29, 2019 · 243344a · 243344a
1 parent 5bfb4c4
commit 243344a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -11,11 +11,11 @@
 from utils.utils import *
 
 # Hyperparameters: train.py --evolve --epochs 2 --img-size 320, Metrics: 0.204      0.302      0.175      0.234 (square smart)
-hyp = {'giou': 0.01,  # giou loss gain
+hyp = {'giou': 0.02,  # giou loss gain
        'xy': 0.2,  # xy loss gain
        'wh': 0.1,  # wh loss gain
-       'cls': 0.03,  # cls loss gain
-       'conf': 2.0,  # conf loss gain
+       'cls': 0.04,  # cls loss gain
+       'conf': 4.5,  # conf loss gain
        'iou_t': 0.5,  # iou target-anchor training threshold
        'lr0': 0.001,  # initial learning rate
        'lrf': -4.,  # final learning rate = lr0 * (10 ** lrf)