mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
b72d5a3f9a
fixed error "dll load failed" on some systems Expanded eyebrows line of face masks. It does not affect mask of FAN-x converter mode.
35 lines
941 B
Python
35 lines
941 B
Python
from pathlib import Path
|
|
|
|
'''
|
|
You can implement your own SampleGenerator
|
|
'''
|
|
class SampleGeneratorBase(object):
|
|
|
|
|
|
def __init__ (self, samples_path, debug, batch_size):
|
|
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 __iter__(self):
|
|
#implement your own iterator
|
|
return self
|
|
|
|
def __next__(self):
|
|
#implement your own iterator
|
|
return None
|