python下载图片保存时,自动以指定比例保存到训练、验证、测试目录中

前言

datasets 包的 load_dataset 方法可以直接从目录下加载数据集并分成训练、验证
和测试集,如下:

from datasets import load_dataset

# 加载数据集
ds = load_dataset('imagefolder', data_dir=r'E:\数据集\mydatasets2333')
print(ds)
# DatasetDict({
   
#     train: Dataset({
   
#         features: ['image', 'label'],

你可能感兴趣的:(python高级,Python,python,机器学习,深度学习)