torchvision使用笔记

1.torchvision.datasets.ImageFolder(root,transform=None,target_transform=None,loader=default_loader)

        是一种组织数据集的方法,把存在文件夹里的数据转化成用来训练和测试的数据集，

root参数是地址，必须写到分类文件夹的上一层，transform是图片转换形式，target_transform :对label进行变换，loader: 指定加载图片的函数，默认操作是读取PIL image对象

 train_data_transforms=transforms.Compose([transforms.RandomResizedCrop(224),
                                              transforms.RandomHorizontalFlip(),
                                              transforms.ToTensor(),
                                              transforms.Normalize([0.485,0.456,0.406],0.229,0.224,0.225])])

train_data_dir="10_monkey_species/training/training"#下次层一定是分类文件夹

train_data=ImageFolder(train_data_dir,transform=train_data_transforms)
    train_data_loader=Data.DataLoader(train_data,batch_size=32,shuffle=True,num_workers=2)