pytorch中数据增强-2

train_transform = transforms.Compose([
    transforms.Resize((224, 224)),

    # 1 FiveCrop
    # transforms.FiveCrop(112),  # 返回的是tuple元组，下面是对元组的转化
    # transforms.Lambda(lambda crops: torch.stack([(transforms.ToTensor()(crop)) for crop in crops])),

    # 2 TenCrop
    transforms.TenCrop(112, vertical_flip=True),  # vertical_flip=False===>水平翻转
    transforms.Lambda(lambda crops: torch.stack([(transforms.ToTensor()(crop)) for crop in crops])),

])

注意：data=[inputs, labels]

inputs.shape是一个5D张量