mirror of https://github.com/kubeflow/examples.git
Update data.py
This commit is contained in:
parent
f86fd5f0fc
commit
43f6475f1e
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue