rename samples to samplelib

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

View file

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