github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

RecycleGAN fixes

Browse source
This commit is contained in:
iperov 2018-12-28 23:13:29 +04:00
parent 1f86d7f1dd
commit 474fff248f
3 changed files with 25 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

4
nnlib/nnlib.py
View file

@ -551,8 +551,8 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
for i in range(n_blocks):
x = ResnetBlock(ngf*4)(x)
x = ReLU()(XNormalization(Conv2DTranspose(ngf*2, 3, 2, 'same')(x)))
x = ReLU()(XNormalization(Conv2DTranspose(ngf , 3, 2, 'same')(x)))
x = ReLU()(XNormalization(PixelShuffler()(Conv2D(ngf*2 *4, 3, 1, 'same')(x))))
x = ReLU()(XNormalization(PixelShuffler()(Conv2D(ngf *4, 3, 1, 'same')(x))))
x = ReflectionPadding2D((3,3))(x)
x = Conv2D(output_nc, 7, 1, 'valid')(x)