Skip to content
Snippets Groups Projects
Commit e154fba1 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Fix the biogenerator when the multiple samples is False

parent a67951e4
No related branches found
No related tags found
1 merge request!79Add keras-based models, add pixel-wise loss, other improvements
......@@ -56,15 +56,20 @@ class BioGenerator(Generator):
self.load_data = load_data
self.biofile_to_label = biofile_to_label
def reader(f):
def _reader(f):
label = int(self.biofile_to_label(f))
data = self.load_data(self.database, f)
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:
yield (d, label, key)
else:
yield (data, label, key)
else:
def reader(f):
return _reader(f)
super(BioGenerator, self).__init__(
biofiles, reader, multiple_samples=multiple_samples, **kwargs
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment