У меня есть несколько папок в каталоге, и в каждой папке сотни изображений (.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
Надеюсь, вы можете мне помочь, большое спасибо!