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)
1条答案
按热度按时间9rbhqvlz1#
嗯,这是我能做到的一种方式。