| 194 | |
| 195 | |
| 196 | class SimpleResNetEncoder4(nn.Module): |
| 197 | def __init__(self, pretrain=False): |
| 198 | super().__init__() |
| 199 | self.resnet_fpn = resnet_fpn_backbone('resnet34', pretrained=pretrain, |
| 200 | norm_layer=None, returned_layers=[1, 2], |
| 201 | trainable_layers=5) |
| 202 | |
| 203 | def forward(self, x): |
| 204 | x = self.resnet_fpn(x) |
| 205 | return x['0'] |
| 206 | |
| 207 | |
| 208 | def conv2d( |
nothing calls this directly
no outgoing calls
no test coverage detected