# 1.导入依赖包
from PIL import Image
import matplotlib.pyplot as plt
from torchvision.transforms import RandomHorizontalFlip, RandomVerticalFlip, RandomCrop, Resize, RandomRotation
from torchvision.transforms import ColorJitter,Compose

# 2.读取图像 PIL--pillow----Image.open() [C,H,W]
img = Image.open('dog.jpg')
plt.imshow(img)
plt.show()

# 3.几何增强
# 3.1 随机水平翻转
HF = RandomHorizontalFlip(p=0.8)
hf_img = HF(img)
plt.imshow(hf_img)
# plt.show()
# 3.2 随机垂直翻转
VF = RandomVerticalFlip(p=0.8)
VF_img = VF(img)
plt.imshow(VF_img)
# plt.show()
# 3.3 随机裁剪
Crop = RandomCrop((224, 224))
crop_img = Crop(img)
plt.imshow(crop_img)
# plt.show()
# 3.4 随机旋转
RR = RandomRotation(45)
RR_img = RR(img)
plt.imshow(RR_img)
# plt.show()
# 3.5 尺寸调整
Resize = Resize((224, 224))
Resize_img = Resize(img)
plt.imshow(Resize_img)
# plt.show()

# 4.颜色增强
CJ = ColorJitter(brightness=0.5, contrast=0.5,saturation=0.5, hue=0.5)
CJ_img = CJ(img)
plt.imshow(CJ_img)
# plt.show()


# 5.组合增强（几何+颜色）
compose = Compose([VF, RR, Resize, HF, CJ])
compose_img = compose(img)
plt.imshow(compose_img)
plt.show()