(self, x, aspp_lo=None, aspp_attn=None, scale_float=None)
| 462 | out_ch=1) |
| 463 | |
| 464 | def _fwd(self, x, aspp_lo=None, aspp_attn=None, scale_float=None): |
| 465 | _, _, final_features = self.backbone(x) |
| 466 | attn = self.scale_attn(final_features) |
| 467 | pred = self.cls_head(final_features) |
| 468 | attn = scale_as(attn, x) |
| 469 | pred = scale_as(pred, x) |
| 470 | |
| 471 | return pred, attn, None, None |
| 472 | |
| 473 | |
| 474 | def HRNet(num_classes, criterion, s2s4=None): |
nothing calls this directly
no test coverage detected