Skip to content

Commit 0a84bc9

Browse files
wuxun-zhangzhreshold
authored andcommitted
make default value of clip aligned with mxnet (#1026)
1 parent c977180 commit 0a84bc9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gluoncv/nn/coder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -251,14 +251,14 @@ class NormalizedBoxCenterDecoder(gluon.HybridBlock):
251251
----------
252252
stds : array-like of size 4
253253
Std value to be divided from encoded values, default is (0.1, 0.1, 0.2, 0.2).
254-
clip : float, default is None
254+
clip : float, default is -1.0
255255
If given, bounding box target will be clipped to this value.
256256
convert_anchor : boolean, default is False
257257
Whether to convert anchor from corner to center format.
258258
259259
"""
260260

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

0 commit comments

Comments
 (0)