torchvision中的数据集使用

1.数据集:

  • 自定义数据集
  • transforms中的类

如何将数据集和transforms结合在一起?

以CIFAR10为列

2.CIFAR10数据集的下载与导入

python 复制代码
import torchvision

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

如果下载比较慢,可以把下载链接放到迅雷中进行下载。后创建dataset文件夹,将下载的数据集放入,即可运行。

导入的datasets和之前讲解的Dataset类是很相似的,实现了__getitem__()方法和__len__()方法。

3.将CIFAR10数据集的图片转换成tensor类型

python 复制代码
import torchvision
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=False,transform=dataset_transform,download=True)

如果tensorboard的step不全可能是因为没加writer.close()

4.torchvision中的其他数据集

按住Ctrl键再点击可以查看源代码,找到url链接之后可以使用迅雷下载。

相关推荐
万邦科技Lafite5 小时前
京东按图搜索京东商品(拍立淘) API (.jd.item_search_img)快速抓取数据
开发语言·前端·数据库·python·电商开放平台·京东开放平台
丁浩6665 小时前
Python机器学习---6.集成学习与随机森林
python·随机森林·机器学习
charlie1145141916 小时前
现代 Python 学习笔记:Statements & Syntax
笔记·python·学习·教程·基础·现代python·python3.13
课堂随想8 小时前
vscode关闭自动激活conda环境
vscode·conda
麦麦大数据9 小时前
F036 vue+flask中医热性药知识图谱可视化系统vue+flask+echarts+mysql
vue.js·python·mysql·flask·可视化·中医中药
移远通信9 小时前
MQTT协议:物联网时代的通信革命
python·物联网·网络协议
Amo Xiang9 小时前
JavaScript逆向与爬虫实战——基础篇(css反爬之动态字体实现原理及绕过)
爬虫·python·js逆向·动态字体
编程让世界美好10 小时前
选手评分问题(python)
python
java1234_小锋10 小时前
PyTorch2 Python深度学习 - PyTorch2安装与环境配置
开发语言·python·深度学习·pytorch2
CClaris10 小时前
深度学习——反向传播的本质
人工智能·python·深度学习