我在一个目录中有多个文件夹，每个文件夹中有数百个图像（.jpg）。

现在，我尝试创建一个csv文件，其中包括图像的名称，逗号和标签。

所以我的文件夹是

狗 猫 鼠 大象

，其中包含标题为dog1.jpg，dog2.jpg，... cat1.jpg等的图像。

我已经尝试使用以下代码列出所有图片名称：

import os

path = '/content../'

files = []
for  r,d,f in os.walk(path):
    for file in f:
        if '.jpg' in file:
          print(file)

这会打印我的图像列表。 我的想法是我创建

import csv 

# csv file name 
with open('test.csv')
category []
if filename == 'dog' then append 0 
if filename == 'cat' then append 1 ... 

但是我无处可去。

最后，csv文件应如下所示：

dog1.jpg,0
dog2.jpg,0
cat1.jpg,1
mouse2.jpg,2
elepant.jpg,3

希望您能帮助我，非常感谢！