山东省建设安全生产协会网站,免费的html5模板,免费关键词搜索引擎工具,wordpress 首页进不去torchvision中的数据集使用 1. torchvision中的数据集使用官网文档注意点1 totensor实例化不要忘记加括号注意点2 download可以一直保持为True代码执行结果 2. DataLoader的使用 1. torchvision中的数据集使用
官网文档
注意左上角的版本 https://pytorch.org/vision/0.9/ 注… torchvision中的数据集使用 1. torchvision中的数据集使用官网文档注意点1 totensor实例化不要忘记加括号注意点2 download可以一直保持为True代码执行结果 2. DataLoader的使用 1. torchvision中的数据集使用
官网文档
注意左上角的版本 https://pytorch.org/vision/0.9/ 注意点1 totensor实例化不要忘记加括号
totensor实例化不要忘记加括号否则后面用数据集序列号的时候会报错
注意点2 download可以一直保持为True
download可以一直保持为True下载一次后指定目录下有下载好的数据集代码不会重复下载也可以自己把下载好的数据集压缩包放到指定目录代码会自动解压缩
代码
from torch.utils.tensorboard import SummaryWriter
from torchvision import datasets, transforms# 用法1
# 数据下载很慢的话 可以使用迅雷下载属性里面可以看到迅雷是从多方下载的速度比较快 https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
train_set datasets.CIFAR10(root./dataset, trainTrue, downloadTrue)
test_set datasets.CIFAR10(root./dataset, trainFalse, downloadTrue)
# 下载的数据集是图片类型可以debug查看数据
print(test_set[0]) # __getitem__ return img, target
print(type(test_set[0]))
img, target test_set[0]
print(target)
print(test_set.classes[target])
print(img)
# PIL 图片可以直接show函数展示
img.show()# 用法2
# 将数据集批量调用transforms使用tensor数据类型
# trans_compose transforms.Compose([transforms.ToTensor]) # 错误写法 会导致后面报错
trans_compose transforms.Compose([transforms.ToTensor()])
train_set2 datasets.CIFAR10(root./dataset, trainTrue, transformtrans_compose, downloadTrue)
test_set2 datasets.CIFAR10(root./dataset, trainFalse, transformtrans_compose, downloadTrue)
print(type(test_set2[2]))
img, target test_set2[0]
print(target)
print(test_set2.classes[target])
print(type(img))
writer SummaryWriter(logs)
for i in range(10):img_tensor, target test_set2[i]writer.add_image(tensor dataset, img_tensor, i)
writer.close()执行结果 p11_torchvision_dataset.py
Files already downloaded and verified
Files already downloaded and verified
(PIL.Image.Image image modeRGB size32x32 at 0x1CF47DA9E20, 3)
class tuple
3
cat
PIL.Image.Image image modeRGB size32x32 at 0x1CF47DA9E20
Files already downloaded and verified
Files already downloaded and verified
class tuple
3
cat
class torch.TensorProcess finished with exit code 02. DataLoader的使用
