From eac988d861adfaeb90d939adffe53dfee96de7df Mon Sep 17 00:00:00 2001 From: Martin Rajchl Date: Sun, 17 Jun 2018 12:41:06 +0100 Subject: [PATCH] fix: issue #27 --- examples/applications/MRBrainS13_tissue_segmentation/reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/applications/MRBrainS13_tissue_segmentation/reader.py b/examples/applications/MRBrainS13_tissue_segmentation/reader.py index 89aaebab..a42a3ec0 100644 --- 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}