从一个目录中读取多个图像并将其转换为.csv文件

import os
import csv
import cv2

def convert_images_to_csv(directory, csv_file):
    # 获取目录下的所有文件和子目录
    for root, dirs, files in os.walk(directory):
        # 遍历所有文件
        for file in files:
            # 检查文件扩展名，确保它是图像文件
            if file.endswith('.jpg') or file.endswith('.png'):
                # 获取图像文件的完整路径
                image_path = os.path.join(root, file)
                
                # 读取图像数据
                image_data = cv2.imread(image_path)
                
                # 将图像数据转换为适当的格式（这里仅示例，可以根据需要进行更多处理）
                image_data = image_data.flatten()  # 将多维数组转换为一维数组
                
                # 打开.csv文件，以追加模式写入图像信息
                with open(csv_file, 'a', newline='') as file:
                    writer = csv.writer(file)
                    writer.writerow([image_path] + list(image_data))

# 示例用法
directory = 'path/to/images'  # 指定图像目录的路径
csv_file = 'images.csv'  # 指定输出的.csv文件名

# 创建.csv文件的表头
with open(csv_file, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Image Path', 'Image Data'])  # 根据需要修改列名

# 调用函数进行转换
convert_images_to_csv(directory, csv_file)