[python]PILで画像を一括保存

Question

[python]PILで画像を一括保存

以下のようなコードを書きカレントディレクトリの直下にあるimagesというフォルダの.png画像をすべてloadすることにしました。(imagesのフォルダには.pngしかないためos.listdirだけで済ませております。)

import numpy as np
import os
from PIL import Image

png_name = os.listdir(path='images')

def name2image(png_name):
    img_data = list()
    for i in png_name:
        img = Image.open('images\'+i)
        img_data.append(img)
    return img_data

img_data = name2image(png_name)

imgはimages内の.png画像がすべてリストの要素になるようなコードでこのimgを再び保存することができれば今後簡単にloadすることができると考えています。
しかしながら私のパソコンのスペックが低いせいで10000枚程度の画像に対して行っているのですが、一時間程度やっても1000枚ぐらいしかappendされません。

MNISTのように60000枚程度の画像が一瞬でloadすることができるのにも関わらずこのように時間がかかってしますのはコードが上手くないからであろうと思います。(MNISTはもともとnumpy配列だから早いだけかもしれませんが、、)
私のコードではなく行いたいことを実行するようにはどのような変更点が求められると思いますか？

詳しい方がいれば教えて頂けると嬉しいです。
よろしくお願いいたします。

Comments

[tkymtmt]

loadしたい画像のサイズはどのくらいですか？

[yu__]

256×256×3です。

Answer 1

pythonで空のlistに大量の要素をappendしようとすると、
要素数が一定数を超えるたびに新しいメモリ領域を確保しようとしてしまうので、
それで時間がかかってしまってるのかもしれません。

以下の例のようにリスト内法表記でlistの初期化と同時に画像をloadするのはどうでしょうか。

def name2image(png_name):
    img_data = [Image.open('images\' + i) for i in png_name]
    return img_data