1. torchvision
torchvision是pytorch工程的一部分,主要用于视觉方面的一个包,包括流行的数据集、模型架构和用于计算机视觉的常见图像转换torchvision.transforms
torchvision官网页面(从pytorch官网docs点开)
2. torchvision.datasets常见的数据集

3. 数据集的使用——以CIFAR10为例
import torchvision
# CIFAR10数据集包括60000张32×32的彩色图片,属于10个类型
from torch.utils.tensorboard import SummaryWriter
dataset_transform = torchvision.transforms

本文介绍了torchvision库在PyTorch中的作用,特别关注其在计算机视觉任务中的数据集处理,如CIFAR10。通过示例展示了如何下载、解压并应用数据转换,同时利用tensorboard展示数据集样本。
最低0.47元/天 解锁文章
3905

被折叠的 条评论
为什么被折叠?



