复制子文件夹下的数据集到新的文件夹，并进行排序

import os
import shutil

# 定义文件路径和新文件夹路径
input_dir = 'F:/tusimple/clips/0313-1'
output_dir = 'F:/新建文件夹'

# 定义计数器
counter = 1

# 遍历文件夹
for root, dirs, files in os.walk(input_dir):
    for file_name in files:
        # 只处理图片文件
        if file_name.endswith('.jpg') or file_name.endswith('.png'):
            # 获取文件的绝对路径
            file_path = os.path.join(root, file_name)
            # 拷贝文件到新文件夹下，并按顺序重新命名
            new_file_name = f'{counter}.jpg'
            new_file_path = os.path.join(output_dir, new_file_name)
            shutil.copyfile(file_path, new_file_path)
            # 计数器自增
            counter += 1