pytorch数据扩增

from torchvision import datasets,transforms
import matplotlib.pylab as plt
import torch

path2data = "./data"
# loading MNIST training dataset
train_data = datasets.MNIST(path2data,train=True,download=False)

# define transformations
data_transform = transforms.Compose([transforms.RandomHorizontalFlip(p=1),
                                     transforms.RandomVerticalFlip(p=1),
                                     transforms.ToTensor(),])

# get a sample image from training dataset
img = train_data[1][0]
print(train_data[1][0])
img_tr = data_transform(img)
img_tr_np = img_tr.numpy()

# show original and transformed images
plt.subplot(121)
plt.imshow(img,cmap="gray")
plt.title("original")
plt.subplot(122)
plt.imshow(img_tr_np[0],cmap="gray")
plt.title("transformed")
plt.show()

data_transform = transforms.Compose([transforms.RandomHorizontalFlip(1),
                                     transforms.RandomVerticalFlip(1),
                                     transforms.ToTensor()])

train_data = datasets.MNIST(path2data,train=True,download=False,transform=data_transform)

结果：

<PIL.Image.Image image mode=L size=28x28 at 0x1D9911C50B8>

本文地址：https://blog.csdn.net/qq_28368377/article/details/107427965