用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

# 创建子文件