import os
import sys
import csv
import pandas as pd
f = open('index.csv','w',encoding='utf-8')
csv_writer = csv.writer(f)
files = os.listdir(r"/home/shuaishuai/文档/实验数据/行为识别/snippets/images")
files.sort()
# print(files)
for file in files: # 注意此处file不是一个list,故不能对files进行排序
path = os.path.join("/home/shuaishuai/文档/实验数据/行为识别/snippets/images", file)
# print(path)
files1 = os.listdir(path)
# print(files)
files1.sort()
for file1 in files1:
# print(path+"/"+file1)
path1 = os.path.join(path+"/"+file1)
# print(path1)
# files2 = os.listdir(path1)
# files2.sort()
# print(files2)
for filepath, dirnames, filenames in os.walk(path1):
# print(filepath)
# print(dirnames)
filenames.sort()
for filename in filenames:
total_path = os.path.join(path1, filename)
label = file
data = pd.DataFrame(columns=('total_path', 'label'))
data = data.append([{'total_path':total_path,'label':label}],ignore_index=True)
data.to_csv('index.csv', index=False, mode='a', header=False)
# print(file)
# print(total_path)
# 此处dirnames,为文件所在文件夹目录
# 此处filenames,为文件名
# dirnames.sort()
# print(dirnames)
# filepath.sort()
# filenames.sort()
# print(filenames)
# dirnames.sort()
# print(dirnames)
# print(total_path)
# for dirname in dirnames:
# for filename in filenames:
# path1 = os.path.join(filepath, dirname)
# for filepath2, dirname2, filename3 in os.walk(filepath+filename):
# dirname2.sort()
# filename2.sort()
# csv_writer.writerow([file, "/home/shuaishuai/文档/实验数据/行为识别/snippets/images"+file+filename+filename2])
# /home/shuaishuai/文档/实验数据/行为识别/snippets/images/ApplyEyeMakeup/v_ApplyEyeMakeup_g01_c01_000000000000_rendered.jpg
# /home/shuaishuai/文档/实验数据/行为识别/snippets/images/ApplyEyeMakeup/v_ApplyEyeMakeup_g01_c01/v_ApplyEyeMakeup_g01_c01_000000000000_rendered.jpg