读取文件夹内的数据

读取文件夹内的数据

from PIL import Image
import os
import numpy as np

# 递归函数，用于获取文件夹内所有PNG图片的文件名
def get_png_files(folder_path):
    png_files = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith('.png'):
                png_files.append(os.path.join(root, file))
    return png_files

# 图片文件夹路径
folder_path = 'E:\subject\pycharm_subject\\bp_cnn_recurent\mnist_train'

# 获取文件夹内所有PNG图片的文件名
png_files = get_png_files(folder_path)

# 读取每张PNG图片，并将其转换为NumPy数组
image_array_list = []
for file_path in png_files:
    # 使用Pillow库加载图片
    image = Image.open(file_path)
    # 将图片转换为NumPy数组，并添加到列表中
    image_array_list.append(np.array(image))

# 将列表中的NumPy数组堆叠为一个三维数组（或四维数组，取决于图片的通道数）
images_array = np.stack(image_array_list)

# 打印数组的形状
print("图像数组的形状:", images_array.shape)