用Python将文件夹中的图片拆分为训练集、验证集、测试集

在 Python 中可以使用 os 模块和 shutil 模块来实现将文件夹中的图片拆分为训练集、验证集、测试集。

示例代码如下：

import os
import shutil

# 源文件夹路径
src_folder = 'path/to/src/folder'
# 目标文件夹路径
dst_folder = 'path/to/dst/folder'
# 文件夹中文件的数量
n_files = len(os.listdir(src_folder))

# 设置比例
train_ratio = 0.6
val_ratio = 0.2
test_ratio = 0.2

# 创建子文件夹
os.makedirs(os.path.join(dst_folder, 'train'))
os.makedirs(os.path.join(dst_folder, 'val'))
os.makedirs(os.path.join(dst_folder, 'test'))

# 拆分文件
for i, file in enumerate(os.listdir(src_folder)):
    src_file = os.path.join(src_folder, file)
    if i < train_ratio * n_files:
        dst_file = os.path.join(dst_folder, 'train', file)
    elif i < (train_ratio + val_ratio) * n_files:
        dst_file = os.path.join(dst_folder, 'val', file)
    else:
        dst_file = os.path.join(dst_folder, 'test', file)
    shutil.copy(src_file, dst_file)

该代码将文件夹中的文件按照设置的比例分别拷贝到目标文件夹的子文件夹中。

可以根据自己需求更改文件夹路径、比例等。