fix train diff for fairmot (PaddlePaddle#3436)

heavengate · Jun 21, 2021 · a8015e4 · a8015e4
1 parent 35bd189
commit a8015e4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
@@ -2015,12 +2015,12 @@ def apply(self, sample, context=None):
 
 @register_op
 class AugmentHSV(BaseOperator):
-    def __init__(self, fraction=0.50, is_bgr=False):
+    def __init__(self, fraction=0.50, is_bgr=True):
         """ 
         Augment the SV channel of image data.
         Args:
-            fraction (float): the fraction for augment 
-            is_bgr (bool): whether the image is BGR mode
+            fraction (float): the fraction for augment. Default: 0.5.
+            is_bgr (bool): whether the image is BGR mode. Default: True.
         """
         super(AugmentHSV, self).__init__()
         self.fraction = fraction

diff --git a/ppdet/modeling/layers.py b/ppdet/modeling/layers.py
@@ -163,6 +163,8 @@ def __init__(self,
                 bias_attr=True,
                 lr_scale=dcn_lr_scale,
                 regularizer=dcn_regularizer,
+                dcn_bias_regularizer=dcn_regularizer,
+                dcn_bias_lr_scale=dcn_lr_scale,
                 skip_quant=skip_quant)
 
         norm_lr = 0. if freeze_norm else 1.