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

4 篇文章 0 订阅
4 篇文章 0 订阅
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

 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值