创建一个带有图像的.csv

Question

假设给我一个训练文件夹。训练文件夹包含有单独的文件夹，其中有图像（CNV、DME、DRUSEN 和 NORMAL 都是检测类型）。

我想检查每个文件夹中的每张图像，并给每张图像分配一个... ...

• 独特的ID 和a 疾病名称

然后让它写入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)

我还没有测试过，但你有这个想法，对吗？