使用图像创建 .csv

Question

假设我得到了一个培训文件夹。训练文件夹包含具有图像的单独文件夹（CNV、DME、DRUSEN 和 NORMAL 都是检测类型）：

我想遍历每个文件夹中的每张图片并为每张图片分配一个

唯一 ID 和 疾病名称

然后让它写入 CSV

我正在尝试完成的示例：

我想要一个 CSV，其中包含我拥有的所有 60,000 张图像的唯一 ID，这些图像包含在一列的文件夹中，疾病类型（或文件：CNV、DME、DRUSEN、NORMAL）在其他列。

Answer 1

您可以在 os.listdir() 的帮助下完成此操作。

csv = 'id, disease\n'
for disease in os.listdir(root_data_path):
    for file in os.listdir(os.path.join(root_data_path, disease)):
        csv += f'{file},{disease}\n'

with open('diseases_ids.csv', 'w+') as output_csv:
    output_csv.write(csv)

我还没有测试过，但你明白了，对吧？

使用图像创建 .csv

Creating a .csv with images

python

csv

machine-learning

image-recognition

deep-learning