从两个不同的文件夹中读取图像，并在 csv 文件中列出带有名称的图像目录答案

【问题标题】：Read images from two different folders and list the image directories with name in a csv file从两个不同的文件夹中读取图像，并在 csv 文件中列出带有名称的图像目录
【发布时间】：2021-06-13 08:05:59
【问题描述】：

有两个图像文件夹名称：bad_pictures 和 good_pictures。两个文件夹代表两个不同的类。我正在尝试编写一个 python 脚本来制作一个包含两列的 CSV/JSON 文件，一列是 CONTENT（图像的完整路径），另一列是 LABEL（坏或好)。喜欢：

据我所知，我们可以从一个文件夹中获取图像：

from PIL import Image
import glob
image_list = []
for filename in glob.glob('yourpath/*.jpg'): 
    im=Image.open(filename)
    image_list.append(im)

这与我的目标相去甚远。我希望您能帮助我们找出实现目标的代码。

【问题讨论】：

这里为什么需要PIL？你有n = 2 标签，每个标签的所有图像都在它们自己的目录中。因此，您只需要枚举每个目录中的文件（例如通过pathlib）并生成路径和标签（这是路径的函数）。
由于您只想存储图像路径和标签，因此您可能不需要读取图像，只需尝试读取这两列的文件名和目录名
@Hetal 和 crcvd：谢谢你们的 cmets，可以分享一下代码吗？
@TazimTaz 请检查发布的解决方案
@Hetal：非常感谢。

标签： python image file image-processing

【解决方案1】：

请查看下方生成 csv 文件：

import csv
import glob
path = "d:/so/good_pictures/*.*"
gp=glob.glob(path)
path = "d:/so/bad_pictures/*.*"
bp=glob.glob(path)
with open('test2.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(["CONTENT", "LABEL"])
    for element in gp:
        writer.writerow([element, "good"])
    for element in bp:
        writer.writerow([element, "bad"])

【讨论】：