Skip to content

Commit

Permalink
make default value of clip aligned with mxnet (dmlc#1026)
Browse files Browse the repository at this point in the history
  • Loading branch information
wuxun-zhang authored and zhreshold committed Nov 5, 2019
1 parent c977180 commit 0a84bc9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions gluoncv/nn/coder.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,14 +251,14 @@ class NormalizedBoxCenterDecoder(gluon.HybridBlock):
----------
stds : array-like of size 4
Std value to be divided from encoded values, default is (0.1, 0.1, 0.2, 0.2).
clip : float, default is None
clip : float, default is -1.0
If given, bounding box target will be clipped to this value.
convert_anchor : boolean, default is False
Whether to convert anchor from corner to center format.
"""

def __init__(self, stds=(0.1, 0.1, 0.2, 0.2), convert_anchor=False, clip=None):
def __init__(self, stds=(0.1, 0.1, 0.2, 0.2), convert_anchor=False, clip=-1.0):
super(NormalizedBoxCenterDecoder, self).__init__()
assert len(stds) == 4, "Box Encoder requires 4 std values."
self._stds = stds
Expand Down

0 comments on commit 0a84bc9

Please sign in to comment.