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

fixed error "Failed to get convolution algorithm" on some systems

Browse source 
fixed error "dll load failed" on some systems
Expanded eyebrows line of face masks. It does not affect mask of FAN-x converter mode.
This commit is contained in:
iperov 2019-08-11 11:17:22 +04:00
commit b72d5a3f9a
15 changed files with 367 additions and 222 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/ModelBase.py
View file

@ -466,7 +466,7 @@ class ModelBase(object):
return imagelib.equalize_and_stack_square (images)
def generate_next_sample(self):
return [next(generator) for generator in self.generator_list]
return [ generator.generate_next() for generator in self.generator_list]
def train_one_iter(self):
sample = self.generate_next_sample()

1
models/Model_RecycleGAN/Model.py
View file

@ -378,7 +378,6 @@ class RecycleGANModel(ModelBase):
return x
return func
nnlib.UNet = UNet
@staticmethod
def UNetTemporalPredictor(output_nc, use_batch_norm, ngf=64, use_dropout=False):

6
models/Model_SAE/Model.py
View file

@ -545,6 +545,12 @@ class SAEModel(ModelBase):
return Norm(norm)( Act(act) (Conv2D(dim, kernel_size=5, strides=2, padding=padding)(x)) )
return func
SAEModel.downscale = downscale
#def downscale (dim, padding='zero', norm='', act='', **kwargs):
# def func(x):
# return BlurPool()( Norm(norm)( Act(act) (Conv2D(dim, kernel_size=5, strides=1, padding=padding)(x)) ) )
# return func
#SAEModel.downscale = downscale
def upscale (dim, padding='zero', norm='', act='', **kwargs):
def func(x):