PythonにおけるNumPyでの画像読み込みについて

Question

いつもお世話になっております。Pythonを始めたばかりの未熟者ですが、よろしくお願いいたします。

Pythonでのnpy読み込みに関するお話なのですが、複数枚画像が存在する際に、一枚ずつ読み込む方法を模索しています。

import numpy as np

Input = np.load('Input_Data.npy')
print('Input_shape = ',Input.shape) #Input_shape = (画像枚数、チャネル、画像の縦サイズ、画像の横サイズ) = (640,1,50,50) 

上記の方法では、640枚の画像としていっぺんに変数Inputに格納されますが、これを一枚ずつ読み込む方法はないかと考えています。あるいは実験的に、一枚だけ読み込んでみる方法でも構いません。他にも、まずは単純に640枚読み込んだあと、一枚ずつ抽出する方法があるのではとも考えています。単純な問題で恐縮ですが、情報をお持ちの方がいらっしゃったらどうぞよろしくお願い致します。

Answer 1

numpy.load()にmmap_modeという引数があります。これを使うとファイルが仮想アドレス空間にマッピングされますので、実行時のオーバーヘッドは避けられると思います。

Input = np.load('Input_Data.npy', mmap_mode='r')

参考URL

• https://numpy.org/doc/stable/reference/generated/numpy.load.html

Comments

[T_Endo]

ご回答ありがとうございます！tiitoiさんの回答をベストアンサーとさせていただきましたが、yymtさんのご意見も参考にさせていただきます。

Answer 2

640枚の画像としていっぺんに変数Inputに格納されますが、これを一枚ずつ読み込む方法はないかと考えています。あるいは実験的に、一枚だけ読み込んでみる方法でも構いません。

640枚がすべて1つの npy ファイルとして保存されている場合、その一部だけ読み込むことはできません。
1枚ずつ読み込みたいのであれば、画像1枚ごとに個別に npy ファイルで保存する必要があります。

他にも、まずは単純に640枚読み込んだあと、一枚ずつ抽出する方法があるのではとも考えています。

for で iterate すれば、1枚ずつ取り出せませんか？

import numpy as np

imgs = np.load("Input_Data.npy")

for img in imgs:
    print(img.shape)  # (チャネル、画像の縦サイズ、画像の横サイズ)

Comments

[T_Endo]

ありがとうございます！無事解決できました。