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

fix train diff for fairmot (#3436)

上级 35bd189b
......@@ -2015,12 +2015,12 @@ class Rbox2Poly(BaseOperator):
@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
......
......@@ -163,6 +163,8 @@ class ConvNormLayer(nn.Layer):
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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册