如何从图像文件夹创建 csv 文件？答案

【问题标题】：how create cvs file from image folders?如何从图像文件夹创建 csv 文件？
【发布时间】：2021-07-12 02:27:20
【问题描述】：

我有两个良性和恶性皮肤癌的图像文件夹，我想获取 CSV 文件，其中包含第一列是图像的路径，第二列是 python 语言的图像标签。我该怎么做？

数据集的路径 '../input/skin-cancer-malignant-vs-benign/train/benign'

'../input/skin-cancer-malignant-vs-benign/train/malignant'

【问题讨论】：

您能否更具体地了解您的需求？什么是“图像的标签”？我认为 os.walk 是此类脚本的一个很好的起点
图片文件的扩展名是什么？即 .png .jpg

标签： python image

【解决方案1】：

查看glob 模块：

benign = glob.glob('{path to benign folder}/*.png')
malignant = glob.glob('{path to malignant folder}/*.png')

这里的 * 仅表示获取此文件夹中所有 .png 文件的文件路径。当然，将 .png 更改为您使用的任何图像格式。

那么就是写数据了

import glob

benign = glob.glob('../input/skin-cancer-malignant-vs-benign/train/benign/*.png')
malignant = glob.glob('../input/skin-cancer-malignant-vs-benign/train/malignant/*.png')

CSV_FILE_NAME = 'my_file.csv'
with open(CSV_FILE_NAME, 'w') as f:
    for path in benign:
        f.write(path)     # write the path in the first column
        f.write(',')      # separate first and second item by a comma
        f.write('benign') # write the label in the second column
        f.write('\n')     # start a new line
    
    for path in malignant:
        f.write(path)
        f.write(',')
        f.write('malignant')
        f.write('\n')

你绝对可以把它写得更简洁，但这更具可读性

【讨论】：