画像はTensorFlowでどうやって読み込ませるの？

2024年6月8日2024年6月17日

画像をTensorFlowで読み込むには、tf.keras.preprocessing.imageモジュールを使用します。以下は、画像を読み込むサンプルコードです。

python
import tensorflow as tf

# 画像を読み込む
def load_image(image_path, target_size):
    img = tf.keras.preprocessing.image.load_img(image_path, target_size=target_size)
    img_array = tf.keras.preprocessing.image.img_to_array(img)
    img_array = img_array / 255.0  # ピクセル値を正規化
    return img_array

# 画像のパス
image_path = "path/to/your/image.jpg"

# 画像のサイズ（読み込む画像のサイズを指定）
target_size = (224, 224)  # 例: (幅, 高さ)

# 画像の読み込み
image = load_image(image_path, target_size)

# 読み込んだ画像の形状を表示
print("Image shape:", image.shape)

このサンプルコードでは、load_image関数を使用して指定された画像を読み込み、指定されたターゲットサイズにリサイズしています。tf.keras.preprocessing.image.load_img関数を使用して画像を読み込み、tf.keras.preprocessing.image.img_to_array関数を使用してNumPy配列に変換しています。その後、ピクセル値を0から1の範囲に正規化しています。