From c8e805667f7915dba92eb8cfd4e4d8fe3d9b6269 Mon Sep 17 00:00:00 2001 From: "Alex J. Champandard" Date: Tue, 8 Nov 2016 17:15:54 +0100 Subject: [PATCH] Fix for padding in convolution. --- enhance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhance.py b/enhance.py index 2dd36a2..1cf570f 100755 --- a/enhance.py +++ b/enhance.py @@ -268,7 +268,7 @@ class Model(object): return list(self.network.values())[-1] def make_layer(self, name, input, units, filter_size=(3,3), stride=(1,1), pad=(1,1), alpha=0.25): - conv = ConvLayer(input, units, filter_size, stride=stride, pad=(0,0), nonlinearity=None) + conv = ConvLayer(input, units, filter_size, stride=stride, pad=pad, nonlinearity=None) prelu = lasagne.layers.ParametricRectifierLayer(conv, alpha=lasagne.init.Constant(alpha)) self.network[name+'x'] = conv self.network[name+'>'] = prelu