Method __init__

network/basic.py:71–82

(self, num_classes, trunk='hrnetv2', criterion=None)

    ASPP-based Segmentation network
    """
    def __init__(self, num_classes, trunk='hrnetv2', criterion=None):
        super(ASPP, self).__init__()
        self.criterion = criterion
        self.backbone, _, _, high_level_ch = get_trunk(trunk)
        self.aspp, aspp_out_ch = get_aspp(high_level_ch,
                                          bottleneck_ch=cfg.MODEL.ASPP_BOT_CH,
                                          output_stride=8)
        self.bot_aspp = nn.Conv2d(aspp_out_ch, 256, kernel_size=1, bias=False)
        self.final = make_seg_head(in_ch=256,
                                   out_ch=num_classes)

        initialize_weights(self.final, self.bot_aspp, self.aspp)

    def forward(self, inputs):
        x = inputs['images']