Skip to content

Commit

Permalink
Fix off-by-one error in BaseDataProvider
Browse files Browse the repository at this point in the history
We already copy the first element to the first position in the array. If we start from zero again, we loose the first image, and pull one more element than actually needed.
  • Loading branch information
nicolov authored Jan 4, 2017
1 parent 42849c5 commit d08d48e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tf_unet/image_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def __call__(self, n):

X[0] = train_data
Y[0] = labels
for i in range(0, n):
for i in range(1, n):
train_data, labels = self._load_data_and_label()
X[i] = train_data
Y[i] = labels
Expand Down

0 comments on commit d08d48e

Please sign in to comment.