torchvision.transform,DataLoader等数据准备操作
torchvision包包含了目前流行的数据集,模型结构和常用的图片转换工具。
torchvision.datasets包含了MNIST,CIFAR,COCO等常见数据集。
例:dset.CIFAR10(root, train=True, transform=None, target_transform=None, download=False)
dset.CIFAR100(root, train=True, transform=None, target_transform=None, download=False)
参数说明: - root :cifar-10-batches-py的根目录 - train :True = 训练集, False = 测试集- download : True = 从互联上下载数据,并将其放在root目录下。如果数据集已经下载,无操作。torchvision.transform。。。。。

浙公网安备 33010602011771号