Pythonで複数画像データを配列化して保存する方法

Question

Pythonで複数画像データを数値化して配列に保存することを考えています。
1枚の画像データはflatten()関数で1行にして、それを画像枚数分だけ配列に保存したいです。

おそらく、下記流れでコーディングをすると思うのですが、どのように表現をすればまったくわかりません。

1.配列の宣言
2.1枚目画像の読み込み
3.画像データを1行にして配列の1行目に保存
4.2枚目画像の読み込み
5.画像データを1行にして配列の2行目に保存
・・・

ご存知の方がいましたら是非、教えていただけると大変助かります。
よろしくお願いいたします。

Comments

[Zuishin]

数値化する必要ありますか？

[退会済みユーザー]

詳細は不明ですが「Deep Learningの教師データとするためのインプット情報」を作っておられるそうで、画像をRGB配列化したものが必要なのだとか。（URL:https://teratail.com/questions/77937）

Answer 1

# coding: utf-8

import glob
from PIL import Image

files = glob.glob('folder/*.png')

for file in files:
    array = []
    img = Image.open(file).convert('RGB')
    w, h = img.size

    for y in range(h):
      for x in range(w):
        array += img.getpixel((x, y))

    f = open('test.txt', 'a')
    out = map(str, array)
    f.write(','.join(out) + '\n')
    f.close()

参考にしたページ
Python3: 指定パスのファイル一覧を取得する
Pythonでリストの要素を一括で変換(文字列などに)

Python初心者ですが考えてみました。

Answer 2

flatten関数についてはわからないので、質問とは違う回答になりますが
もしバッチ処理をしたいのだとするのであれば、違うやり方ですが少し前に書いたコードがあるので考え方の参考にでもなればと思います

from PIL import Image
import numpy as np
import os

#画像フォルダを開く
images = os.listdir("ディレクトリ")

#バッチを収納する為のリストを作成する
batch_list = []

#開いたフォルダの中にある画像の名前をイテレーションで全て抽出する
for i in images:
    #listdirで開くと画像の名前とは関係のないThumbs.dbが抽出されるので無視する
    if i == "Thumbs.db":
        continue
    #imagesの中にある全ての画像の配列を開いていく
    image = np.array(Image.open("ディレクトリ/"+i))
    #reshapeを使って開かれた配列を1次元配列に変換する
    image_risize = image.reshape(image.size)
    #バッチリストに追加していく
    batch_list.append(image_risize)

print(len(batch_list))
print(len(batch_list[0]))

実行結果
354
320000

私の環境では500*640の画像がフォルダの中に354個ありましたので
望み通りバッチ処理が出来ているのがわかるかと思います