Python实现从Dataset文件夹生成分块帧路径CSV数据集

代码实现 #

import os
import csv
from pathlib import Path

# 定义根文件夹路径
root_dir = "Dataset"
# 输出CSV文件路径
output_csv = "frame_groups.csv"

# 获取所有子文件夹
subfolders = [f.path for f in os.scandir(root_dir) if f.is_dir()]

with open(output_csv, mode='w', newline='', encoding='utf-8') as csv_file:
    writer = csv.writer(csv_file)
    
    for folder in subfolders:
        # 获取文件夹内所有图像文件，按名称排序（确保连续帧顺序正确）
        # 可根据实际图像格式调整后缀，比如添加'.png'等
        image_files = sorted([
            str(Path(folder) / img) 
            for img in os.listdir(folder) 
            if img.lower().endswith(('.jpg', '.jpeg', '.png', '.bmp'))
        ])
        
        # 按7个一组分块写入CSV
        for i in range(0, len(image_files), 7):
            group = image_files[i:i+7]
            writer.writerow(group)

关键说明 #

• 路径处理：使用pathlib模块拼接路径，确保跨平台兼容性
• 排序图像：对图像文件按名称排序，保证连续帧的顺序正确，避免因系统文件排序导致帧顺序混乱
• 分块逻辑：利用Python切片i:i+7实现每7个一组的分块，最后一组自动包含剩余的不足7个的帧
• 图像格式适配：代码中默认支持常见的图像格式，可根据实际情况修改endswith中的后缀列表

内容的提问来源于stack exchange，提问作者SAUMYA BHANDARY