CortexFoundation / StyleTransferTrilogy

风格迁移三部曲
517 stars 124 forks source link

keras ReflectPadding2D has problem? #8

Open pageedward opened 5 years ago

pageedward commented 5 years ago

util.py line 130 .132 reverse1 = K.reverse(inputs,-1) inputs = K.concatenate([reverse1[:,:,:,-self.padding_length:],inputs,reverse1[:,:,:,:self.padding_length]],axis = -1) reverse2 = K.reverse(inputs,-2) inputs = K.concatenate([reverse2[:,:,-self.padding_length:,:],inputs,reverse2[:,:,:self.padding_length,:]],axis = -2) maybe it is K.concatenate([reverse1[:,:,:,:self.padding_length],inputs,reverse1[:,:,:,-self.padding_length:]],axis = -1) @ypwhs