介绍
文档
可以去看官方文档![[pytorch入门] 4. torchvision中数据集的使用,# pytorch,深度学习,pytorch,人工智能,python](https://imgs.yssmx.com/Uploads/2024/01/817737-1.png)
可以在里面找到一些数据集的使用![[pytorch入门] 4. torchvision中数据集的使用,# pytorch,深度学习,pytorch,人工智能,python](https://imgs.yssmx.com/Uploads/2024/01/817737-2.png)
文章来源：https://www.toymoban.com/news/detail-817737.html
CIFAR10
import torchvision
from torch.utils.tensorboard import SummaryWriter
dataset_transform = torchvision.transforms.Compose([
torchvision.transforms.ToTensor(),
])
train_set = torchvision.datasets.CIFAR10(root = "./dataset",train=True, download=True)
test_set = torchvision.datasets.CIFAR10(root = "./dataset",train=False, download=True)
writer = SummaryWriter('logs')
for i in range(10):
img, target = train_set[i]
writer.add_image('train_set', img, i)
writer.close()
这就可以在项目目录下看到下载的压缩文件，若文件已经下载好，再次运行时不会重复下载
文件下载下来比较慢，可以复制链接用迅雷等工具下文章来源地址https://www.toymoban.com/news/detail-817737.html
到了这里，关于[pytorch入门] 4. torchvision中数据集的使用的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
