Update data.py

This commit is contained in:
Rebecca McFadden 2019-09-03 17:44:06 -05:00
parent f86fd5f0fc
commit 43f6475f1e
1 changed files with 1 additions and 1 deletions

View File

@ -73,7 +73,7 @@ def walk_images(path, image_size=160):
# write out good images # write out good images
imgs.append(image) imgs.append(image)
except img.shape[2] != 3: except img.shape[2] != 3:
print('{}\n{}\n'.format(e, image)) print('{}\n'.format(image))
return imgs return imgs