python - 如何使用图像列表中的标签创建 csv 文件

Question

我在一个目录中有多个文件夹，每个文件夹中有数百张图像(.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

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

Answer 1

这是我的看法

带有说明的文件: https://hastebin.com/hopeneluhi.py

请注意我没有对此进行测试。如果有任何问题请告诉我:)