我们想要把图像的短边调整为256,然后随机裁剪(randomcrop)
为224大小的正方形。也就是说,我们打算组合一个Rescale
和 RandomCrop
的变换。 我们可以调用一个简单的类 torchvision.transforms.Compose
来实现这一操作。具体实现如下图:
scale = Rescale(256)
crop = RandomCrop(128)
composed = transforms.Compose([Rescale(256),
RandomCrop(224)])
# 在样本上应用上述的每个变换。
fig = plt.figure()
sample = face_dataset[65]
for i, tsfrm in enumerate([scale, crop, composed]):
transformed_sample = tsfrm(sample)
ax = plt.subplot(1, 3, i + 1)
plt.tight_layout()
ax.set_title(type(tsfrm).__name__)
show_landmarks(**transformed_sample)
plt.show()