fix: issue #27

DLTK · Jun 17, 2018 · eac988d · eac988d
1 parent b44cf72
commit eac988d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/applications/MRBrainS13_tissue_segmentation/reader.py b/examples/applications/MRBrainS13_tissue_segmentation/reader.py
@@ -84,7 +84,7 @@ def _augment(img, lbl):
                 n_examples=n_examples,
                 classes=9)
 
-            for e in range(n_examples):
+            for e in range(len(images)):
                 yield {'features': {'x': images[e].astype(np.float32)},
                        'labels': {'y': lbl[e].astype(np.int32)},
                        'subject_id': subject_id}