图片数据集生成csv文件

将图片数据集生成CSV文件的步骤如下：

1. 导入必要的库，如`os`，`csv`和`PIL`。

2. 确定数据集的路径和文件名，并使用`os.listdir()`函数读取所有图像文件的名称。

3. 创建一个CSV文件并编写标题行，例如`"filename","label"`。

4. 遍历数据集中的图像文件，使用PIL库读取每个图像的标签，并将其写入CSV文件中。

下面是一个示例代码，可以将数据集路径和文件名按实际情况修改：

import os
import csv
from PIL import Image

# 数据集路径和文件名
data_path = "dataset/"
csv_file = "dataset.csv"

# 获取所有图像文件名
img_files = os.listdir(data_path)

# 创建CSV文件并编写标题行
with open(csv_file, mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(["filename", "label"])

    # 遍历数据集中的图像文件，将文件名和标签写入CSV文件
    for img_file in img_files:
        img_path = os.path.join(data_path, img_file)
        label = img_file.split("_")[0]
        writer.writerow([img_file, label])

这样就能生成一个名为`dataset.csv`的CSV文件，其中包含所有图像文件的文件名和标签。