CortexFoundation / StyleTransferTrilogy

风格迁移三部曲

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

keras ReflectPadding2D has problem?

pageedward opened this issue · comments

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