torchvision中数据集的使用
创始人
2024-04-06 04:31:19
0

一、pytorch官网学习

1.1 模块

  • 处理语音
  • 核心模块
  • 处理文本
  • 处理视觉图像的

1.2 torchvision

  • tensorboard
  • tochvision

就是来源于这里
在这里插入图片描述

1.2.1 dataset模块

在这里插入图片描述

1.2.2 io模块

1.3 提供常规的神经网络


二、【实验1】导入cifar数据集并输出部分信息

2.1 代码

import torchvision# 使用torchvision自带的数据集
train_set = torchvision.datasets.CIFAR10(root='./dataset', train=True, download=True)
test_set = torchvision.datasets.CIFAR10(root='./dataset', train=False, download=True)# 将下载好的测试集中的东西打印出来
print(test_set[0])
print(test_set.classes)img, target = test_set[0]
print(img)
print(target)
print(test_set.classes[target])img.show()

2.2 结果

这里可以看到img,和target 的值是获取到了训练集中这个大类中对应的目标值
在这里插入图片描述

三、【实验2】使用循环将图片列表进行输出

tensor数据类型的话就可以使用tensorboard进行输出使用

如何使用writer进行输出?

数据集和transform结合在一起

3.1 代码

import torchvision# 使用transforms转换数据集
from torch.utils.tensorboard import SummaryWriterdataset_trans = torchvision.transforms.Compose([torchvision.transforms.ToTensor()
])# 使用torchvision自带的数据集
train_set = torchvision.datasets.CIFAR10(root='./dataset', train=True, transform=dataset_trans, download=True)
test_set = torchvision.datasets.CIFAR10(root='./dataset', train=False, transform=dataset_trans, download=True)# # 将下载好的测试集中的东西打印出来
# print(test_set[0])
# print(test_set.classes)
#
# img, target = test_set[0]
# print(img)
# print(target)
# print(test_set.classes[target])
# img.show()# 使用writer将获取到的图片进行输出
writer = SummaryWriter('p10')
for i in range(10):img, target = test_set[i]writer.add_image('test_set', img, i)writer.close()

D:\ANACONDA\envs\pytorch\python.exe C:/Users/Administrator/Desktop/Code/learn_pytorch/P14_dataset_transforms.py
Files already downloaded and verified
Files already downloaded and verified

Process finished with exit code 0

3.2 结果

来吧展示
在这里插入图片描述

四、如何下载标准的数据集

在官网中寻找自己想要的东西和答案

五、尽可能的进行实验


相关内容

热门资讯

喜欢穿一身黑的男生性格(喜欢穿... 今天百科达人给各位分享喜欢穿一身黑的男生性格的知识,其中也会对喜欢穿一身黑衣服的男人人好相处吗进行解...
发春是什么意思(思春和发春是什... 本篇文章极速百科给大家谈谈发春是什么意思,以及思春和发春是什么意思对应的知识点,希望对各位有所帮助,...
网络用语zl是什么意思(zl是... 今天给各位分享网络用语zl是什么意思的知识,其中也会对zl是啥意思是什么网络用语进行解释,如果能碰巧...
为什么酷狗音乐自己唱的歌不能下... 本篇文章极速百科小编给大家谈谈为什么酷狗音乐自己唱的歌不能下载到本地?,以及为什么酷狗下载的歌曲不是...
家里可以做假山养金鱼吗(假山能... 今天百科达人给各位分享家里可以做假山养金鱼吗的知识,其中也会对假山能放鱼缸里吗进行解释,如果能碰巧解...
华为下载未安装的文件去哪找(华... 今天百科达人给各位分享华为下载未安装的文件去哪找的知识,其中也会对华为下载未安装的文件去哪找到进行解...
四分五裂是什么生肖什么动物(四... 本篇文章极速百科小编给大家谈谈四分五裂是什么生肖什么动物,以及四分五裂打一生肖是什么对应的知识点,希...
怎么往应用助手里添加应用(应用... 今天百科达人给各位分享怎么往应用助手里添加应用的知识,其中也会对应用助手怎么添加微信进行解释,如果能...
客厅放八骏马摆件可以吗(家里摆... 今天给各位分享客厅放八骏马摆件可以吗的知识,其中也会对家里摆八骏马摆件好吗进行解释,如果能碰巧解决你...
苏州离哪个飞机场近(苏州离哪个... 本篇文章极速百科小编给大家谈谈苏州离哪个飞机场近,以及苏州离哪个飞机场近点对应的知识点,希望对各位有...