mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
dc11ec32be
fixed model sizes from previous update. avoided bug in ML framework(keras) that forces to train the model on random noise. Converter: added blur on the same keys as sharpness Added new model 'TrueFace'. This is a GAN model ported from https://github.com/NVlabs/FUNIT Model produces near zero morphing and high detail face. Model has higher failure rate than other models. Keep src and dst faceset in same lighting conditions.
39 lines
1 KiB
Python
39 lines
1 KiB
Python
from pathlib import Path
|
|
|
|
'''
|
|
You can implement your own SampleGenerator
|
|
'''
|
|
class SampleGeneratorBase(object):
|
|
|
|
|
|
def __init__ (self, samples_path, debug=False, batch_size=1):
|
|
if samples_path is None:
|
|
raise Exception('samples_path is None')
|
|
|
|
self.samples_path = Path(samples_path)
|
|
self.debug = debug
|
|
self.batch_size = 1 if self.debug else batch_size
|
|
self.last_generation = None
|
|
self.active = True
|
|
|
|
def set_active(self, is_active):
|
|
self.active = is_active
|
|
|
|
def generate_next(self):
|
|
if not self.active and self.last_generation is not None:
|
|
return self.last_generation
|
|
self.last_generation = next(self)
|
|
return self.last_generation
|
|
|
|
#overridable
|
|
def get_total_sample_count(self):
|
|
return 0
|
|
|
|
#overridable
|
|
def __iter__(self):
|
|
#implement your own iterator
|
|
return self
|
|
|
|
def __next__(self):
|
|
#implement your own iterator
|
|
return None
|