(self, num_branches, blocks, num_blocks, num_inchannels,
num_channels, fuse_method, multi_scale_output=True)
| 89 | |
| 90 | class HighResolutionModule(nn.Module): |
| 91 | def __init__(self, num_branches, blocks, num_blocks, num_inchannels, |
| 92 | num_channels, fuse_method, multi_scale_output=True): |
| 93 | super(HighResolutionModule, self).__init__() |
| 94 | self._check_branches( |
| 95 | num_branches, blocks, num_blocks, num_inchannels, num_channels) |
| 96 | |
| 97 | self.num_inchannels = num_inchannels |
| 98 | self.fuse_method = fuse_method |
| 99 | self.num_branches = num_branches |
| 100 | |
| 101 | self.multi_scale_output = multi_scale_output |
| 102 | |
| 103 | self.branches = self._make_branches( |
| 104 | num_branches, blocks, num_blocks, num_channels) |
| 105 | self.fuse_layers = self._make_fuse_layers() |
| 106 | self.relu = nn.ReLU(True) |
| 107 | |
| 108 | def _check_branches(self, num_branches, blocks, num_blocks, |
| 109 | num_inchannels, num_channels): |
nothing calls this directly
no test coverage detected