(self, x)
| 106 | self.dropout = nn.Dropout(0.5) |
| 107 | |
| 108 | def forward(self, x): |
| 109 | x1 = self.aspp1(x) |
| 110 | x_aspp_deforms = [aspp_deform(x) for aspp_deform in self.aspp_deforms] |
| 111 | x5 = self.global_avg_pool(x) |
| 112 | x5 = F.interpolate(x5, size=x1.size()[2:], mode='bilinear', align_corners=True) |
| 113 | x = torch.cat((x1, *x_aspp_deforms, x5), dim=1) |
| 114 | |
| 115 | x = self.conv1(x) |
| 116 | x = self.bn1(x) |
| 117 | x = self.relu(x) |
| 118 | |
| 119 | return self.dropout(x) |
nothing calls this directly
no outgoing calls
no test coverage detected