(self, num_branches, blocks, num_blocks, num_inchannels,
num_channels, fuse_method, multi_scale_output=True)
| 100 | |
| 101 | class HighResolutionModule(nn.Module): |
| 102 | def __init__(self, num_branches, blocks, num_blocks, num_inchannels, |
| 103 | num_channels, fuse_method, multi_scale_output=True): |
| 104 | super(HighResolutionModule, self).__init__() |
| 105 | self._check_branches( |
| 106 | num_branches, blocks, num_blocks, num_inchannels, num_channels) |
| 107 | |
| 108 | self.num_inchannels = num_inchannels |
| 109 | self.fuse_method = fuse_method |
| 110 | self.num_branches = num_branches |
| 111 | |
| 112 | self.multi_scale_output = multi_scale_output |
| 113 | |
| 114 | self.branches = self._make_branches( |
| 115 | num_branches, blocks, num_blocks, num_channels) |
| 116 | self.fuse_layers = self._make_fuse_layers() |
| 117 | self.relu = nn.ReLU(True) |
| 118 | |
| 119 | def _check_branches(self, num_branches, blocks, num_blocks, |
| 120 | num_inchannels, num_channels): |
nothing calls this directly
no test coverage detected