如何将图像文件转换为带标签的CSV

import os
import pandas as pd
from PIL import Image

# 图像文件夹路径和标签文件路径
image_folder = 'path/to/image/folder'
label_file = 'path/to/label/file.csv'

# 获取图像文件列表
image_files = [f for f in os.listdir(image_folder) if f.endswith('.jpg') or f.endswith('.png')]

# 创建空的DataFrame用于存储数据
data = pd.DataFrame(columns=['image_path', 'label'])

# 遍历图像文件列表
for image_file in image_files:
    # 图像路径
    image_path = os.path.join(image_folder, image_file)
    
    # 读取图像文件
    image = Image.open(image_path)
    
    # 进行图像预处理和特征提取
    # ...
    
    # 获取图像标签
    label = get_label(image_file)
    
    # 将图像路径和标签添加到DataFrame中
    data = data.append({'image_path': image_path, 'label': label}, ignore_index=True)

# 保存DataFrame为CSV文件
data.to_csv(label_file, index=False)