Skip to content

Commit

Permalink
**add sigmoid after mask**
Browse files Browse the repository at this point in the history
  • Loading branch information
CharlesShang committed Dec 5, 2018
1 parent 41ee18d commit b648fdd
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions dcn_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,5 +64,6 @@ def forward(self, input):
out = self.conv_offset_mask(input)
o1, o2, mask = torch.chunk(out, 3, dim=1)
offset = torch.cat((o1, o2), dim=1)
mask = torch.sigmoid(mask)
func = DCNv2Function(self.stride, self.padding, self.dilation, self.deformable_groups)
return func(input, offset, mask, self.weight, self.bias)

0 comments on commit b648fdd

Please sign in to comment.