(self, input)
| 22 | nn.init.constant_(self.conv_offset_mask.bias, 0) |
| 23 | |
| 24 | def forward(self, input): |
| 25 | offset_mask = self.conv_offset_mask(input) |
| 26 | offset_x, offset_y, mask = torch.chunk(offset_mask, 3, dim=1) |
| 27 | offset = torch.cat((offset_x, offset_y), dim=1) |
| 28 | mask = mask.sigmoid() |
| 29 | return super(DLADeformConv, self).forward(input, offset, mask) |
| 30 | |
| 31 | |
| 32 | class DeformConvWithOff(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected