Проблема, с которой вы сталкиваетесь, заключается в том, что os.listdir()
выдаст вам список файлов в каталоге, который вы используете, без относительного пути к ним. Это оставляет вам несколько вариантов.
Опция A:
Предварительно укажите путь к файлу, а именно:
for filename in os.listdir(path):
if filename.endswith(".png"):
im = image.imread(path+"/"+filename)
...
Опция B: glob
from glob import glob
for filename in glob(path+"/*.png"):
im = image.imread(filename)
...