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'],

猜你喜欢

转载自blog.csdn.net/weixin_43721000/article/details/129837126