Method __init__

network/basic.py:42–49

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

    Basic segmentation network, no ASPP, no Mscale
    """
    def __init__(self, num_classes, trunk='hrnetv2', criterion=None):
        super(Basic, self).__init__()
        self.criterion = criterion
        self.backbone, _, _, high_level_ch = get_trunk(
            trunk_name=trunk, output_stride=8)
        self.seg_head = make_seg_head(in_ch=high_level_ch,
                                      out_ch=num_classes)
        initialize_weights(self.seg_head)

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