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

有两个图像文件夹名称：坏图片和好图片。两个文件夹代表两个不同的类。我正在尝试编写一个python脚本来生成一个CSV/JSON文件，该文件将包含两列，一列是内容（图像的完整路径），另一列是标签（坏或好）。比如：

据我所知，我们可以通过以下方式从单个文件夹获取图像：

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

---

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

请检查以下内容以生成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"])

---

为什么您需要

PIL

在这里？您有

n=2

标签，每个标签的所有图像都位于各自的目录中。因此，您只需枚举每个目录中的文件（例如，通过

pathlib

）并生成路径和标签（这是路径的函数）。由于您只想存储图像路径和标签，因此可能不需要读取图像，只需读取这两个文件的文件名和目录名columns@Hetal和crcvd：谢谢你的评论，你能分享代码吗？@TazimTaz请查看发布的内容solution@Hetal：非常感谢。