Итерировать каждое изображение в каталоге - PullRequest
Новая
       24

Итерировать каждое изображение в каталоге

0 голосов
/ 01 августа 2020

Я пытаюсь запустить приведенный ниже код и вызываю функцию из другого ноутбука, но, похоже, не могу понять, как загрузить все данные изображения в моем каталоге. другая записная книжка, которую я импортировал. 

def ela_form(path, quality):

    im_name = path
    rename_image = im_name.split('.')[0] + '.resaved.jpg'
    ELA_filename = im_name.split('.')[0] + '.ela.png'

    im = Image.open(im_name).convert('RGB')
    im.save(rename_image, 'JPEG', quality=quality)
    resaved_im = Image.open(rename_image)

    ela_image = ImageChops.difference(im, resaved_im)

    extrema = ela_image.getextrema()
    max_diff = max([ex[1] for ex in extrema])
    if max_diff == 0:
       max_diff = 1
   scale = 255.0 / max_diff

   ela_image = ImageEnhance.Brightness(ela_image).enhance(scale)

   return ela_image

Мои изображения хранятся в 'dataset \ Training_Data_Preprocessed'

Dataset.csv хранится в 'dataset \ Dataset.csv 

> Error:FileNotFoundError                         Traceback (most recent
> call last) <ipython-input-16-920134a3a471> in <module>
>       3 for index, row in dataset.iterrows():
>       4 
> ----> 5     x.append(array(ela_form(row[1], 90).resize((128, 128))).flatten() / 255.0)
>       6     y.append(row[0])
> 
> ~\Desktop\Jupyter Notebooks\ELA.ipynb in ela_form(path, quality)
> 
> D:\Anaconda\lib\site-packages\PIL\Image.py in open(fp, mode)    2807  
> 2808     if filename:
> -> 2809         fp = builtins.open(filename, "rb")    2810         exclusive_fp = True    2811 
> 
> FileNotFoundError: [Errno 2] No such file or directory: 'im1_o.bmp'
...