(self, x)
| 255 | self.rebnconv1d = REBNCONV(mid_ch * 2, out_ch, dirate=1) |
| 256 | |
| 257 | def forward(self, x): |
| 258 | |
| 259 | hx = x |
| 260 | |
| 261 | hxin = self.rebnconvin(hx) |
| 262 | |
| 263 | hx1 = self.rebnconv1(hxin) |
| 264 | hx = self.pool1(hx1) |
| 265 | |
| 266 | hx2 = self.rebnconv2(hx) |
| 267 | hx = self.pool2(hx2) |
| 268 | |
| 269 | hx3 = self.rebnconv3(hx) |
| 270 | |
| 271 | hx4 = self.rebnconv4(hx3) |
| 272 | |
| 273 | hx3d = self.rebnconv3d(torch.cat((hx4, hx3), 1)) |
| 274 | hx3dup = _upsample_like(hx3d, hx2) |
| 275 | |
| 276 | hx2d = self.rebnconv2d(torch.cat((hx3dup, hx2), 1)) |
| 277 | hx2dup = _upsample_like(hx2d, hx1) |
| 278 | |
| 279 | hx1d = self.rebnconv1d(torch.cat((hx2dup, hx1), 1)) |
| 280 | |
| 281 | return hx1d + hxin |
| 282 | |
| 283 | |
| 284 | ### RSU-4F ### |
nothing calls this directly
no test coverage detected