mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-03-12 20:42:45 -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.
40 lines
1.0 KiB
Python
40 lines
1.0 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
|