(self, x,stage=0)
| 424 | |
| 425 | |
| 426 | def forward(self, x,stage=0): |
| 427 | conv0 = self.conv0(x) |
| 428 | conv2 = self.conv2(self.conv1(conv0)) |
| 429 | conv4 = self.conv4(self.conv3(conv2)).squeeze(2) |
| 430 | x=self.conv6(self.conv5(conv4)) |
| 431 | x=conv4+self.conv7(x) |
| 432 | x=x.unsqueeze(2) |
| 433 | x = conv2 + self.conv9(x) |
| 434 | x = conv0 + self.conv11(x) |
| 435 | x = self.prob(x) |
| 436 | return x |
| 437 | class AggWeightNetVolume(nn.Module): |
| 438 | def __init__(self, in_channels=32,hid_channels=1,out_channels=1,relu=True): |
| 439 | super(AggWeightNetVolume, self).__init__() |
nothing calls this directly
no outgoing calls
no test coverage detected