如何在Python中生成多个32x32灰度图像的平均图像？

import numpy as np
import cv2
import os

def read_gray_image(file_path):
    image = cv2.imread(file_path, cv2.IMREAD_GRAYSCALE)
    return image

def generate_average_image(image_folder):
    image_files = os.listdir(image_folder)
    image_count = len(image_files)
    average_image = np.zeros((32, 32), dtype=np.float32)

    for file in image_files:
        image_path = os.path.join(image_folder, file)
        image = read_gray_image(image_path)
        average_image += image.astype(np.float32) / image_count

    average_image = average_image.astype(np.uint8)
    return average_image

image_folder = 'path/to/image/folder/'
average_image = generate_average_image(image_folder)
cv2.imwrite('average_image.png', average_image)