dataset与DataLoader的简单使用

        利用torchvision下载给定的数据到dataset中，并采用DataLoader进行加载。实例代码如下：

#dataset与dataLoader一些使用
#数据来源于torchvision中的数据
import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

#获得dataset
dataset_transfrom=torchvision.transforms.Compose(
    [torchvision.transforms.ToTensor()])
train_dataset=torchvision.datasets.CIFAR10(root='./dataset',train=True,transform=dataset_transfrom,download=True)
test_dataset=torchvision.datasets.CIFAR10(root='./dataset',train=False,transform=dataset_transfrom,download=True)

#设置batch_size
batchsize=64    

#设置DataLoader
train_loader=DataLoader(dataset=train_dataset,batch_size=batchsize,shuffle=True)
test_loader=DataLoader(dataset=test_dataset,batch_size=batchsize)

#采用Tensorboard进行展示,因为训练数据较多，这里对测试数据进行展示
writer=SummaryWriter(log_dir='logs')
i=0
for datas in test_loader:
    imgs,targets=datas
    writer.add_images(tag="test_img", img_tensor=imgs,global_step=i)
    i+=1
writer.close()

           Tensorboard展示如下：