Commit e154fba1 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Fix the biogenerator when the multiple samples is False

parent a67951e4
...@@ -56,15 +56,20 @@ class BioGenerator(Generator): ...@@ -56,15 +56,20 @@ class BioGenerator(Generator):
self.load_data = load_data self.load_data = load_data
self.biofile_to_label = biofile_to_label self.biofile_to_label = biofile_to_label
def reader(f): def _reader(f):
label = int(self.biofile_to_label(f)) label = int(self.biofile_to_label(f))
data = self.load_data(self.database, f) data = self.load_data(self.database, f)
key = str(f.make_path("", "")).encode("utf-8") key = str(f.make_path("", "")).encode("utf-8")
if self.multiple_samples: return data, label, key
if multiple_samples:
def reader(f):
data, label, key = _reader(f)
for d in data: for d in data:
yield (d, label, key) yield (d, label, key)
else: else:
yield (data, label, key) def reader(f):
return _reader(f)
super(BioGenerator, self).__init__( super(BioGenerator, self).__init__(
biofiles, reader, multiple_samples=multiple_samples, **kwargs biofiles, reader, multiple_samples=multiple_samples, **kwargs
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment