mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
25 lines
618 B
Python
25 lines
618 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
|
|
|
|
#overridable
|
|
def __iter__(self):
|
|
#implement your own iterator
|
|
return self
|
|
|
|
def __next__(self):
|
|
#implement your own iterator
|
|
return None
|
|
|