fix WongKinYiu#149

mkrupczak3 · Mar 3, 2024 · 51190cf · 51190cf
1 parent 2004346
commit 51190cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -543,6 +543,7 @@ def main(opt, callbacks=Callbacks()):
             'cls_pw': (1, 0.5, 2.0),  # cls BCELoss positive_weight
             'obj': (1, 0.2, 4.0),  # obj loss gain (scale with pixels)
             'obj_pw': (1, 0.5, 2.0),  # obj BCELoss positive_weight
+            'dfl': (1, 1.0, 1.75), # dfl loss gain
             'iou_t': (0, 0.1, 0.7),  # IoU training threshold
             'anchor_t': (1, 2.0, 8.0),  # anchor-multiple threshold
             'anchors': (2, 2.0, 10.0),  # anchors per output grid (0 to ignore)

diff --git a/train_dual.py b/train_dual.py
@@ -553,6 +553,7 @@ def main(opt, callbacks=Callbacks()):
             'cls_pw': (1, 0.5, 2.0),  # cls BCELoss positive_weight
             'obj': (1, 0.2, 4.0),  # obj loss gain (scale with pixels)
             'obj_pw': (1, 0.5, 2.0),  # obj BCELoss positive_weight
+            'dfl': (1, 1.0, 1.75), # dfl loss gain
             'iou_t': (0, 0.1, 0.7),  # IoU training threshold
             'anchor_t': (1, 2.0, 8.0),  # anchor-multiple threshold
             'anchors': (2, 2.0, 10.0),  # anchors per output grid (0 to ignore)