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链接之后可以使用迅雷下载。

相关推荐
鸡鸭扣1 小时前
Docker:3、在VSCode上安装并运行python程序或JavaScript程序
运维·vscode·python·docker·容器·js
paterWang2 小时前
基于 Python 和 OpenCV 的酒店客房入侵检测系统设计与实现
开发语言·python·opencv
东方佑2 小时前
使用Python和OpenCV实现图像像素压缩与解压
开发语言·python·opencv
神秘_博士3 小时前
自制AirTag,支持安卓/鸿蒙/PC/Home Assistant,无需拥有iPhone
arm开发·python·物联网·flutter·docker·gitee
Moutai码农4 小时前
机器学习-生命周期
人工智能·python·机器学习·数据挖掘
小白教程5 小时前
python学习笔记,python处理 Excel、Word、PPT 以及邮件自动化办公
python·python学习·python安装
武陵悭臾5 小时前
网络爬虫学习:借助DeepSeek完善爬虫软件,实现模拟鼠标右键点击,将链接另存为本地文件
python·selenium·网络爬虫·pyautogui·deepseek·鼠标右键模拟·保存链接为htm
摸鱼仙人~5 小时前
ImportError: cannot import name ‘FixtureDef‘ from ‘pytest‘
conda·pytest·fastapi
代码猪猪傻瓜coding5 小时前
关于 形状信息提取的说明
人工智能·python·深度学习
码界筑梦坊7 小时前
基于Flask的第七次人口普查数据分析系统的设计与实现
后端·python·信息可视化·flask·毕业设计