(self, num_branches, blocks, num_blocks, num_inchannels,
num_channels, fuse_method, multi_scale_output=True)
| 31 | |
| 32 | class HighResolutionModule(nn.Module): |
| 33 | def __init__(self, num_branches, blocks, num_blocks, num_inchannels, |
| 34 | num_channels, fuse_method, multi_scale_output=True): |
| 35 | super(HighResolutionModule, self).__init__() |
| 36 | self._check_branches( |
| 37 | num_branches, blocks, num_blocks, num_inchannels, num_channels) |
| 38 | |
| 39 | self.num_inchannels = num_inchannels |
| 40 | self.fuse_method = fuse_method |
| 41 | self.num_branches = num_branches |
| 42 | |
| 43 | self.multi_scale_output = multi_scale_output |
| 44 | |
| 45 | self.branches = self._make_branches( |
| 46 | num_branches, blocks, num_blocks, num_channels) |
| 47 | self.fuse_layers = self._make_fuse_layers() |
| 48 | self.relu = nn.ReLU(True) |
| 49 | |
| 50 | def _check_branches(self, num_branches, blocks, num_blocks, |
| 51 | num_inchannels, num_channels): |
nothing calls this directly
no test coverage detected