读取文件夹中的图片.png并将其转换为.npy格式

import imageio
import os
import cv2
import numpy as np

folder_path = r"D:\all_data"  # 文件夹路径
output_folder = r"D:\picturre"  # 输出文件夹路径
# 确保输出文件夹存在
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    file_path = os.path.join(folder_path, filename)

    # 判断是否为PNG格式图片,这里可以改为别的图片格式
    if os.path.isfile(file_path) and filename.lower().endswith(".png"):
        # 读取图片并将其转换为灰度图像
        #image = cv2.imread(file_path)      #可以用opencv转化为灰度图
        image = imageio.imread(file_path)   #也可以使用imageio转化为灰度图
        print(image)
        print(image.shape)
        k1 =image[:, :, 0]            #查看rgb图片的三个通道中0通道的数值，和转化无关，可有可无
      
        # 构造输出文件名
        output_filename = os.path.splitext(filename)[0] + ".npy"
        output_path = os.path.join(output_folder, output_filename)

        # 将灰度图像保存为NumPy数组
        np.save(output_path, image)
        print(f"Converted {file_path} to {output_path}")