深度学习：数据加载

最新推荐文章于 2024-07-11 08:19:06 发布

ZERO_pan

最新推荐文章于 2024-07-11 08:19:06 发布

阅读量450

点赞数

分类专栏：人工智能——深度学习

本文链接：https://blog.csdn.net/qq_42911863/article/details/126256390

版权

人工智能——深度学习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

0.前言

框架：pytorch

1.直接从网上下载数据集，用于模型的测试

    tsf = transforms.Compose([transforms.ToTensor()])
    train_data = datasets.FashionMNIST(root='./data/fashionmnist/train',train=True,transform=tsf,download=True)
    test_data = datasets.FashionMNIST(root='./data/fashionmnist/test',train=False,transform=tsf,download=True)
    train_iter = DataLoader(train_data,batch_size=32,shuffle=True)
    test_iter = DataLoader(test_data,batch_size=32,shuffle=False)

2.从本地文件夹中加载

    tsf = transforms.Compose([transforms.Resize((28, 28)),
                              transforms.ToTensor()])
    train_data = datasets.ImageFolder(root=r'image\dogcat\dogcat\train',transform=tsf)
    train_iter = DataLoader(train_data,batch_size=32,shuffle=True)