From 491f306d9796ad38b6e69dfb7d31f4f26e7d9080 Mon Sep 17 00:00:00 2001 From: "Joshua Z. Zhang" Date: Mon, 14 Oct 2019 11:27:31 -0700 Subject: [PATCH] make initial mask centered at 1 rather than 0.5 --- python/mxnet/gluon/contrib/cnn/conv_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mxnet/gluon/contrib/cnn/conv_layers.py b/python/mxnet/gluon/contrib/cnn/conv_layers.py index 3a30f59de44f..098463eca968 100644 --- a/python/mxnet/gluon/contrib/cnn/conv_layers.py +++ b/python/mxnet/gluon/contrib/cnn/conv_layers.py @@ -379,7 +379,7 @@ def hybrid_forward(self, F, x, offset_weight, deformable_conv_weight, offset_bia offset_t = F.slice_axis(offset, axis=1, begin=0, end=18) mask = F.slice_axis(offset, axis=1, begin=18, end=None) - mask = F.sigmoid(mask) + mask = F.sigmoid(mask) * 2 if deformable_conv_bias is None: act = F.contrib.ModulatedDeformableConvolution(data=x, offset=offset_t, mask=mask,