生成csv数据集文件,该文件通过python中的文件夹结构进行解释

ghhaqwfi  于 2023-01-06  发布在  Python
关注(0)|答案(1)|浏览(107)

我有一个文件夹“数据集”,其中包含了一堆文件夹与自己的名字。每个文件夹包含n帧,这是连续的图像。n范围从5-20图像。我需要一个Python代码,其中迭代通过所有文件夹中的“数据集”,并从每个文件夹,将它们分成7个文件块。它将每个文件块的路径写入一行。然后它从同一个文件夹中取出剩余的文件并存储在csv文件的新行中。然后,它将进入数据集类内的另一个文件夹,并对数据集中的所有文件夹执行相同的操作。

9rbhqvlz

9rbhqvlz1#

嗯,这是我能做到的一种方式。

import os
import csv

dataset_path = "Datasets/VOS/train/masks"

with open("data.csv", "w", newline="") as csvfile:

    writer = csv.writer(csvfile)

    for folder in os.listdir(dataset_path):

        folder_path = os.path.join(dataset_path, folder)

        files = os.listdir(folder_path)

        for i in range(0, len(files), 7):
            chunk = files[i:i+7]
            names = [os.path.join(dataset_path, folder, file) for file in chunk]
            writer.writerow(names)

相关问题