天天看点

对图片数据集生成一个索引,写入CSV文件格式(address, label)

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