torchvision.transform,DataLoader等数据准备操作

torchvision包包含了目前流行的数据集,模型结构和常用的图片转换工具。

  1. 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目录下。如果数据集已经下载,无操作。
  2. torchvision.transform。。。。。
posted @ 2021-10-05 10:30  发呆的e  阅读(116)  评论(0)    收藏  举报