diff --git a/train.py b/train.py index b0885308..253f1b18 100644 --- a/train.py +++ b/train.py @@ -28,7 +28,7 @@ hyp = {'giou': 3.54, # giou loss gain 'cls_pw': 1.0, # cls BCELoss positive_weight 'obj': 64.3, # obj loss gain (*=img_size/320 if img_size != 320) 'obj_pw': 1.0, # obj BCELoss positive_weight - 'iou_t': 0.1, # iou training threshold + 'iou_t': 0.20, # iou training threshold 'lr0': 0.01, # initial learning rate (SGD=5E-3, Adam=5E-4) 'lrf': 0.0005, # final learning rate (with cos scheduler) 'momentum': 0.937, # SGD momentum