make initial mask centered at 1 rather than 0.5

apache · Oct 14, 2019 · 491f306 · 491f306
1 parent b7bd5dd
commit 491f306
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,