github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

update FANSeg

Browse source
This commit is contained in:
Colombo 2020-03-08 10:34:48 +04:00
parent 6f4ea69d4d
commit 18d93376fc
4 changed files with 61 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
models/ModelBase.py
View file

@ -412,7 +412,13 @@ class ModelBase(object):
return imagelib.equalize_and_stack_square (images)
def generate_next_samples(self):
self.last_sample = sample = [ generator.generate_next() for generator in self.generator_list]
sample = []
for generator in self.generator_list:
if generator.is_initialized():
sample.append ( generator.generate_next() )
else:
sample.append ( [] )
self.last_sample = sample
return sample
def train_one_iter(self):