def __init__()

in containers/Shoot/CNN/unet.py [0:0]

• 11 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, num_filter,kernel,pad,activation,pool="avg",**kwargs):
        super(Down_Block, self).__init__(**kwargs)
        
        with self.name_scope():
            self.conv1=Conv_Block(num_filter,kernel,pad,activation)
            self.conv1_a=Conv_Block(num_filter,1,0,activation)
            self.conv2=Conv_Block(num_filter,kernel,pad,activation)
            self.conv2_a=Conv_Block(num_filter,1,0,activation)
            if(pool=="avg"):
                self.pool=nn.AvgPool2D(pool_size=(2,2),strides=(2,2))
            else:
                self.pool=nn.MaxPool2D(pool_size=(2,2),strides=(2,2))