transforms.RandomResizedCrop(224)
随机裁剪
transforms.RandomHorizontalFlip()
随机水平翻转
tansforms.ToTensor()
转换成Tensor
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
transforms.Resize((224, 224))
压缩成224*224
os.getcwd()
获取当前文件的根目录
os.path.join('a' + 'b')
将a和b路径连接在一起,用于路径连接
datasets.ImageFolder('path')
加载数据集
train_dataset.class_to_idx
获取类别所对应的索引值