本文介绍: drop_last 当取数据有余数时,是舍去还是保留。出现以上问题需要numberworks设置为0。将该数据数据显示在tensorboard中。加载数据集,查看数据集的属性。将图片转换tensor类型

在这里插入图片描述
使用transform
在这里插入图片描述
加载数据集,查看数据集的属性
在这里插入图片描述
图片转换tensor类型
在这里插入图片描述

dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])

train_set = torchvision.datasets.CIFAR10(root="./dataset",train=True,transform= dataset_transform,download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset",train=True,transform= dataset_transform,download=True)

print(test_set[0])

在这里插入图片描述
将该数据数据显示在tensorboard
Dataloader
在这里插入图片描述
在这里插入图片描述

import torchvision
from torch.utils.data import DataLoader

#准备测试数据集
test_data = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor(),download=True)

test_loader = DataLoader(dataset = test_data,batch_size=4,shuffle=True,num_workers=0,drop_last=False)
#测试数据集中第一张图片
img,target = test_data[0]
print(img.shape)
print(target)

for data in  test_loader:
    imgs,targets = data
    print(imgs.shape)
    print(targets)

在这里插入图片描述出现以上问题需要numberworks设置为0
drop_last 当取数据有余数时,是舍去还是保留
在这里插入图片描述

import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

#准备测试数据集
test_data = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor())

test_loader = DataLoader(dataset = test_data,batch_size=64,shuffle=True,num_workers=0,drop_last=True)
#测试数据集中第一张图片
img,target = test_data[0]
print(img.shape)
print(target)

writer = SummaryWriter("DataLodaer")

#shuffle 为True 两次结果不一样
for epoch in range(2):
    step = 0
    for data in  test_loader:
        imgs,targets = data
        # print(imgs.shape)
        # print(targets)
        writer.add_images("Epoch:{}".format(epoch),imgs,step)
        step = step+1

writer.close()

神经网络
在这里插入图片描述
在这里插入图片描述

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注