(self, x)
| 25 | self.dropout=nn.Dropout2d(0.1) |
| 26 | |
| 27 | def forward(self, x): |
| 28 | y=[x] |
| 29 | for stage in self.stages: |
| 30 | z=stage(x) |
| 31 | z=F.interpolate(z,size=x.shape[-2:],align_corners=False,mode="bilinear") |
| 32 | y.append(z) |
| 33 | x=torch.cat(y,1) |
| 34 | x = self.bottleneck(x) |
| 35 | return x |
| 36 | class AlignedModule(nn.Module): |
| 37 | #SFNet-DFNet |
| 38 | def __init__(self, inplane, outplane): |
nothing calls this directly
no outgoing calls
no test coverage detected