未验证 提交 a8015e4b 编写于 作者: F FlyingQianMM 提交者: GitHub

fix train diff for fairmot (#3436)

上级 35bd189b
...@@ -2015,12 +2015,12 @@ class Rbox2Poly(BaseOperator): ...@@ -2015,12 +2015,12 @@ class Rbox2Poly(BaseOperator):
@register_op @register_op
class AugmentHSV(BaseOperator): 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. Augment the SV channel of image data.
Args: Args:
fraction (float): the fraction for augment fraction (float): the fraction for augment. Default: 0.5.
is_bgr (bool): whether the image is BGR mode is_bgr (bool): whether the image is BGR mode. Default: True.
""" """
super(AugmentHSV, self).__init__() super(AugmentHSV, self).__init__()
self.fraction = fraction self.fraction = fraction
......
...@@ -163,6 +163,8 @@ class ConvNormLayer(nn.Layer): ...@@ -163,6 +163,8 @@ class ConvNormLayer(nn.Layer):
bias_attr=True, bias_attr=True,
lr_scale=dcn_lr_scale, lr_scale=dcn_lr_scale,
regularizer=dcn_regularizer, regularizer=dcn_regularizer,
dcn_bias_regularizer=dcn_regularizer,
dcn_bias_lr_scale=dcn_lr_scale,
skip_quant=skip_quant) skip_quant=skip_quant)
norm_lr = 0. if freeze_norm else 1. norm_lr = 0. if freeze_norm else 1.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册