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

相关推荐
bst@微胖子44 分钟前
Python高级语法之selenium
开发语言·python·selenium
查理零世2 小时前
【蓝桥杯集训·每日一题2025】 AcWing 6118. 蛋糕游戏 python
python·算法·蓝桥杯
魔尔助理顾问3 小时前
一个简洁高效的Flask用户管理示例
后端·python·flask
java1234_小锋3 小时前
一周学会Flask3 Python Web开发-request请求对象与url传参
开发语言·python·flask·flask3
木觞清4 小时前
PyTorch与TensorFlow的对比:哪个框架更适合你的项目?
人工智能·pytorch·tensorflow
诚信爱国敬业友善6 小时前
常见排序方法的总结归类
开发语言·python·算法
架构默片7 小时前
【JAVA工程师从0开始学AI】,第五步:Python类的“七十二变“——当Java的铠甲遇见Python的液态金属
java·开发语言·python
小哥山水之间8 小时前
在 Python 中操作 Excel 文件
开发语言·python·excel
wang_yb10 小时前
『Python底层原理』--CPython的变量实现机制
python·databook
databook10 小时前
『Python底层原理』--CPython的变量实现机制
后端·python