PyTorch基础——使用pytorch加载cifar10数据集

PyTorch基础——使用pytorch加载cifar10数据集使用torchvision.datasets模块可以加载cifar10数据集,涉及函数为torchvision.datasets.CIFAR10(root,train,download)root:cifar10数据集存放目录train:True,表示加载训练数据集,False,表示加载验证数据集download:True,表示cifar10数据集在root指定的文件夹不存在时,会自动下载,False,表示不管root指定文件夹是否存在cifar10数据集,都不会自动下载cifar10数据集

大家好,又见面了,我是你们的朋友全栈君。

使用torchvision.datasets模块可以加载cifar10数据集,涉及函数为torchvision.datasets.CIFAR10(root, train, download)
root: cifar10数据集存放目录
train: True,表示加载训练数据集,False,表示加载验证数据集
download: True,表示cifar10数据集在root指定的文件夹不存在时,会自动下载,False,表示不管root指定文件夹是否存在cifar10数据集,都不会自动下载cifar10数据集
【sample】

from torchvision import datasets
cifar10 = datasets.CIFAR10(root="C:\\cifar10_dataset", train=True, download=True)
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/152083.html原文链接:https://javaforall.net

赞 (1)
全栈程序员-站长的头像全栈程序员-站长

相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号