(self, cfg, **kargs)
| 153 | |
| 154 | class SparseInsDecoder(nn.Module): |
| 155 | def __init__(self, cfg, **kargs) -> None: |
| 156 | super().__init__() |
| 157 | in_channels = cfg.encoder.out_dims + 2 |
| 158 | self.output_iam = cfg.decoder.output_iam |
| 159 | self.scale_factor = cfg.decoder.scale_factor |
| 160 | self.sparse_decoder_weight = cfg.sparse_decoder_weight |
| 161 | self.inst_branch = InstanceBranch(cfg.decoder, in_channels) |
| 162 | # dim, num_convs, kernel_dim, in_channels |
| 163 | self.mask_branch = MaskBranch(cfg.decoder, in_channels) |
| 164 | self.sparse_inst_crit = SparseInstCriterion( |
| 165 | num_classes=cfg.decoder.num_classes, |
| 166 | matcher=SparseInstMatcher(), |
| 167 | cfg=cfg) |
| 168 | self._init_weights() |
| 169 | |
| 170 | def _init_weights(self): |
| 171 | self.inst_branch._init_weights() |
nothing calls this directly
no test coverage detected