Python SDnetDecoderBlock Beispiele

Beispiel #1

    def __init__(self, params):
        super(SDnetConditioner, self).__init__()
        se_block_type = se.SELayer.SSE
        params['num_channels'] = 2
        params['num_filters'] = 16
        self.encode1 = sm.SDnetEncoderBlock(params)

        params['num_channels'] = 16

        self.encode2 = sm.SDnetEncoderBlock(params)

        self.encode3 = sm.SDnetEncoderBlock(params)

        self.encode4 = sm.SDnetEncoderBlock(params)

        self.bottleneck = sm.GenericBlock(params)

        params['num_channels'] = 16

        self.decode1 = sm.SDnetDecoderBlock(params)
        self.channel_conv_d1 = nn.Linear(params['num_filters'], 64, bias=True)

        self.decode2 = sm.SDnetDecoderBlock(params)
        self.channel_conv_d2 = nn.Linear(params['num_filters'], 64, bias=True)

        self.decode3 = sm.SDnetDecoderBlock(params)
        self.channel_conv_d3 = nn.Linear(params['num_filters'], 64, bias=True)

        self.decode4 = sm.SDnetDecoderBlock(params)
        self.channel_conv_d4 = nn.Linear(params['num_filters'], 64, bias=True)

        params['num_channels'] = 16

        self.classifier = sm.ClassifierBlock(params)
        self.sigmoid = nn.Sigmoid()

Beispiel #2

    def __init__(self, params):
        super(SDnetConditioner, self).__init__()
        se_block_type = se.SELayer.SSE
        params['num_channels'] = 2
        params['num_filters'] = 16
        self.encode1 = sm.SDnetEncoderBlock(params)

        params['num_channels'] = 16
        self.encode2 = sm.SDnetEncoderBlock(params)

        self.encode3 = sm.SDnetEncoderBlock(params)

        self.encode4 = sm.SDnetEncoderBlock(params)

        self.bottleneck = sm.GenericBlock(params)
        self.squeeze_conv_bn = nn.Conv2d(in_channels=params['num_filters'],
                                         out_channels=1,
                                         kernel_size=(1, 1),
                                         padding=(0, 0),
                                         stride=1)
        params['num_channels'] = 16
        self.decode1 = sm.SDnetDecoderBlock(params)
        self.decode2 = sm.SDnetDecoderBlock(params)
        self.decode3 = sm.SDnetDecoderBlock(params)
        self.decode4 = sm.SDnetDecoderBlock(params)
        params['num_channels'] = 16
        self.classifier = sm.ClassifierBlock(params)
        self.sigmoid = nn.Sigmoid()

Beispiel #3

Datei: few_shot_segmentor_model3.py Projekt: tanghaotommy/few-shot-segmentation

 def __init__(self, params):
     super(SDnetConditioner, self).__init__()
     params['num_channels'] = 1
     params['num_filters'] = 64
     self.encode1 = sm.SDnetEncoderBlock(params)
     params['num_channels'] = 64
     self.encode2 = sm.SDnetEncoderBlock(params)
     self.encode3 = sm.SDnetEncoderBlock(params)
     self.bottleneck = sm.GenericBlock(params)
     params['num_channels'] = 128
     self.decode1 = sm.SDnetDecoderBlock(params)
     self.squeeze_conv_d1 = nn.Conv2d(in_channels=params['num_filters'], out_channels=1,
                                    kernel_size=(1, 1),
                                    padding=(0, 0),
                                    stride=1)
     self.decode2 = sm.SDnetDecoderBlock(params)
     self.squeeze_conv_d2 = nn.Conv2d(in_channels=params['num_filters'], out_channels=1,
                                    kernel_size=(1, 1),
                                    padding=(0, 0),
                                    stride=1)
     self.decode3 = sm.SDnetDecoderBlock(params)
     self.squeeze_conv_d3 = nn.Conv2d(in_channels=params['num_filters'], out_channels=1,
                                    kernel_size=(1, 1),
                                    padding=(0, 0),
                                    stride=1)
     params['num_channels'] = 64
     self.classifier = sm.ClassifierBlock(params)
     self.sigmoid = nn.Sigmoid()

Beispiel #4

Datei: few_shot_segmentor_shaban_baseline.py Projekt: SaqibMamoon/few-shot-segmentation

    def __init__(self, params):
        super(SDnetConditioner, self).__init__()
        params['num_channels'] = 2
        params['num_filters'] = 16
        self.encode1 = sm.SDnetEncoderBlock(params)

        params['num_channels'] = 16
        self.encode2 = sm.SDnetEncoderBlock(params)

        self.encode3 = sm.SDnetEncoderBlock(params)

        self.encode4 = sm.SDnetEncoderBlock(params)

        self.bottleneck = sm.GenericBlock(params)

        params['num_channels'] = 16
        self.decode1 = sm.SDnetDecoderBlock(params)

        self.decode2 = sm.SDnetDecoderBlock(params)

        self.decode3 = sm.SDnetDecoderBlock(params)

        self.decode4 = sm.SDnetDecoderBlock(params)

        params['num_channels'] = 16
        self.classifier = sm.ClassifierBlock(params)
        self.sigmoid = nn.Sigmoid()

        self.fc_layer = nn.Linear(params['num_filters'], 64, bias=True)

Beispiel #5

 def __init__(self, params):
     super(SDnetSegmentor, self).__init__()
     params['num_channels'] = 1
     params['num_filters'] = 64
     self.encode1 = sm.SDnetEncoderBlock(params)
     params['num_channels'] = 64
     self.encode2 = sm.SDnetEncoderBlock(params)
     self.encode3 = sm.SDnetEncoderBlock(params)
     self.bottleneck = sm.GenericBlock(params)
     params['num_channels'] = 128
     self.decode1 = sm.SDnetDecoderBlock(params)
     self.decode2 = sm.SDnetDecoderBlock(params)
     self.decode3 = sm.SDnetDecoderBlock(params)
     params['num_channels'] = 64
     self.classifier = sm.ClassifierBlock(params)

Beispiel #6

Datei: few_shot_segmentor_feature_fusion_weightRegShaban_baseline.py Projekt: SaqibMamoon/few-shot-segmentation

    def __init__(self, params):
        super(SDnetSegmentor, self).__init__()
        params['num_channels'] = 1
        params['num_filters'] = 64
        self.encode1 = sm.SDnetEncoderBlock(params)
        params['num_channels'] = 64 + 16
        self.encode2 = sm.SDnetEncoderBlock(params)
        self.encode3 = sm.SDnetEncoderBlock(params)
        self.encode4 = sm.SDnetEncoderBlock(params)
        self.bottleneck = sm.GenericBlock(params)

        self.decode1 = sm.SDnetDecoderBlock(params)
        self.decode2 = sm.SDnetDecoderBlock(params)
        self.decode3 = sm.SDnetDecoderBlock(params)
        self.decode4 = sm.SDnetDecoderBlock(params)
        params['num_channels'] = 64
        self.classifier = sm.ClassifierBlock(params)
        self.soft_max = nn.Softmax2d()
        self.sigmoid = nn.Sigmoid()

Beispiel #7

 def __init__(self, params):
     super(SDnetSegmentor, self).__init__()
     se_block_type = se.SELayer.SSE
     params['num_channels'] = 1
     params['num_filters'] = 64
     self.encode1 = sm.SDnetEncoderBlock(params)
     params['num_channels'] = 64
     self.encode2 = sm.SDnetEncoderBlock(params)
     self.encode3 = sm.SDnetEncoderBlock(params)
     self.encode4 = sm.SDnetEncoderBlock(params)
     self.bottleneck = sm.GenericBlock(params)
     params['num_channels'] = 128
     self.decode1 = sm.SDnetDecoderBlock(params)
     self.decode2 = sm.SDnetDecoderBlock(params)
     self.decode3 = sm.SDnetDecoderBlock(params)
     self.decode4 = sm.SDnetDecoderBlock(params)
     params['num_channels'] = 64
     self.classifier = sm.ClassifierBlock(params)
     self.soft_max = nn.Softmax2d()

Beispiel #8

Datei: few_shot_segmentor_feature_fusion_baseline.py Projekt: SaqibMamoon/few-shot-segmentation

    def __init__(self, params):
        super(SDnetConditioner, self).__init__()
        se_block_type = se.SELayer.SSE
        params['num_channels'] = 2
        params['num_filters'] = 16
        self.encode1 = sm.SDnetEncoderBlock(params)
        params['num_channels'] = 16
        self.encode2 = sm.SDnetEncoderBlock(params)

        self.encode3 = sm.SDnetEncoderBlock(params)

        self.encode4 = sm.SDnetEncoderBlock(params)

        self.bottleneck = sm.GenericBlock(params)

        self.decode1 = sm.SDnetDecoderBlock(params)

        self.decode2 = sm.SDnetDecoderBlock(params)

        self.decode3 = sm.SDnetDecoderBlock(params)

        self.decode4 = sm.SDnetDecoderBlock(params)