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

rename samples to samplelib

Browse source
This commit is contained in:
iperov 2019-03-27 10:44:13 +04:00
parent 773e8d80e0
commit 3cf3bb786e
14 changed files with 9 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

24
samples/SampleGeneratorBase.py
View file

@ -1,24 +0,0 @@
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