【Pytorch】5.DataLoder的使用

什么是DataLoader

个人理解是，如果Dataset的所有数据相当于一副扑克牌，DataLoader就相当于从扑克牌中抽取几张，我们可以规定一次抽取的张数，或者以什么规则进行抽取

DataLoader的使用

查阅官网的文档，主要有这几个参数比较常用
在这里插入图片描述
其中dataset可以用上一篇文章来进行创建
具体的实现方法为

import torchvision
from torch.utils.data import DataLoadertest_dataset = torchvision.datasets.CIFAR10(root='./dataset', train=False, transform=torchvision.transforms.ToTensor(), download=True)
# 这里采用测试集是因为测试集较小，运行较快test_dataLoader = DataLoader(test_dataset, batch_size=64, shuffle=False, num_workers=0, drop_last=True)for data in test_dataLoader:            #从test_dataLoader中取出dataimgs, labels = dataprint(labels)

然后我们可以加入tensorBoard可视化处理

import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWritertest_dataset = torchvision.datasets.CIFAR10(root='./dataset', train=False, transform=torchvision.transforms.ToTensor(),download=True)
# 这里采用测试集是因为测试集较小，运行较快test_dataLoader = DataLoader(test_dataset, batch_size=64, shuffle=False, num_workers=0, drop_last=False)writer = SummaryWriter(log_dir='./logs')
i = 0
for data in test_dataLoader:  # 从test_dataLoader中取出dataimgs, labels = dataprint(imgs.shape)writer.add_images('test_loader1', imgs, i)       # 注意这是add_imagesi = i + 1writer.close()