Methods refactoring.

Change-Id: I16b6ae01e70fa130a4df463ee55af97f4752e3b3
zimoqingfeng · Sep 14, 2021 · 7fc8994 · 7fc8994
1 parent b2793ff
commit 7fc8994
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/configs/umop/umop_r50_fpn_1x_coco.py b/configs/umop/umop_r50_fpn_1x_coco.py
@@ -38,20 +38,14 @@
             target_means=[.0, .0, .0, .0],
             target_stds=[0.1, 0.1, 0.2, 0.2]),
         loss_cls=dict(
-            type='FocalLoss',
+            type='ProgressiveFocalLoss',
             use_sigmoid=True,
-            gamma=2.0,
-            alpha=0.25,
+            delta=0.5,
+            w=0.5,
             loss_weight=1.0),
         loss_bbox=dict(type='GIoULoss', loss_weight=2.0),
         loss_centerness=dict(
-            type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0),
-        # Added Loss
-        loss_conv3=dict(type='ProgressiveFocalLoss', use_sigmoid=True, delta=0.5, w=0.5, loss_weight=1.0),
-        loss_conv4=dict(type='ProgressiveFocalLoss', use_sigmoid=True, delta=0.5, w=0.5, loss_weight=1.0),
-        loss_conv5=dict(type='ProgressiveFocalLoss', use_sigmoid=True, delta=0.5, w=0.5, loss_weight=1.0),
-        loss_conv6=dict(type='ProgressiveFocalLoss', use_sigmoid=True, delta=0.5, w=0.5, loss_weight=1.0),
-        loss_conv7=dict(type='ProgressiveFocalLoss', use_sigmoid=True, delta=0.5, w=0.5, loss_weight=1.0)
+            type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0)
     ),
     # training and testing settings
     train_cfg=dict(