(self, x)
| 192 | ) |
| 193 | |
| 194 | def forward(self, x): |
| 195 | x8,x16= x["8"], x["16"] |
| 196 | x16=self.head16(x16) |
| 197 | x8=self.head8(x8) |
| 198 | x16_up = self.fpn_align([x8, x16]) |
| 199 | x8 = x8 + x16_up |
| 200 | x8=self.conv(x8) |
| 201 | x16_up=F.interpolate(x16, x8.shape[-2:], mode="bilinear", align_corners=True) |
| 202 | x8=torch.cat([x8,x16_up],dim=1) |
| 203 | x = self.conv_last(x8) |
| 204 | return x |
| 205 | |
| 206 | class FaPNDecoder(nn.Module): |
| 207 | # FaPN paper |
nothing calls this directly
no outgoing calls
no test coverage detected