diff --git a/samplelib/SampleGeneratorFace.py b/samplelib/SampleGeneratorFace.py index 70f8d71..68229c2 100644 --- a/samplelib/SampleGeneratorFace.py +++ b/samplelib/SampleGeneratorFace.py @@ -142,4 +142,6 @@ class SampleGeneratorFace(SampleGeneratorBase): for i in range(len(x)): batches[i].append ( x[i] ) + filenames.append(sample.filename) + yield ([ np.array(batch) for batch in batches], filenames)