(self, num_branches, blocks, num_blocks, num_inchannels,
num_channels, fuse_method, multi_scale_output=True)
| 107 | |
| 108 | class HighResolutionModule(nn.Module): |
| 109 | def __init__(self, num_branches, blocks, num_blocks, num_inchannels, |
| 110 | num_channels, fuse_method, multi_scale_output=True): |
| 111 | super(HighResolutionModule, self).__init__() |
| 112 | self._check_branches( |
| 113 | num_branches, blocks, num_blocks, num_inchannels, num_channels) |
| 114 | |
| 115 | self.num_inchannels = num_inchannels |
| 116 | self.fuse_method = fuse_method |
| 117 | self.num_branches = num_branches |
| 118 | |
| 119 | self.multi_scale_output = multi_scale_output |
| 120 | |
| 121 | self.branches = self._make_branches( |
| 122 | num_branches, blocks, num_blocks, num_channels) |
| 123 | self.fuse_layers = self._make_fuse_layers() |
| 124 | self.relu = nn.ReLU(False) |
| 125 | |
| 126 | def _check_branches(self, num_branches, blocks, num_blocks, |
| 127 | num_inchannels, num_channels): |
nothing calls this directly
no test coverage detected