padded_convolution2d.py 文件源码

def __call__(self, model):
        if self.stack == 'vertical':
            model = ZeroPadding2D(padding=(self.filter_size[0]//2, 0, self.filter_size[1]//2, self.filter_size[1]//2))(model)
            model = Convolution2D(2*self.filters, self.filter_size[0]//2+1, self.filter_size[1], border_mode='valid')(model)
        elif self.stack == 'horizontal':
            model = ZeroPadding2D(padding=(0, 0, self.filter_size[1]//2, 0))(model)
            if self.mask == 'A':
                model = Convolution2D(2*self.filters, 1, self.filter_size[1]//2, border_mode='valid')(model)
            else:
                model = Convolution2D(2*self.filters, 1, self.filter_size[1]//2+1, border_mode='valid')(model)

        return model