torchvision.transforms

from torchvision import transforms
import cv2

"""
torchvision.transforms常用变换类
transforms.Compose([具体算子，传入类])  组合变换
transforms.Normalize(mean, std),
transforms.Resize(size),
transforms.Scale(size),
transforms.CenterCrop(size),
transforms.RandomCrop(size),
transforms.RandomResizedCrop(size,scale),
transforms.RandomHorizontalFlip(),
transforms.RandomVerticalFlip(),
transforms.RandomRotation(),
transforms.ToTensor(),
transforms.ToPILImage(),
torchvision.transforms编程实战
"""

from PIL import Image
import matplotlib.pyplot as plt

transformer = transforms.Compose([

    # transforms.Resize((1000, 1000)),
    # transforms.Pad(padding=4,
    #                # fill=1,
    #                padding_mode="reflect"),
    transforms.CenterCrop(500),
    # transforms.RandomCrop(1000, pad_if_needed=True, padding_mode='reflect'),
    # transforms.RandomResizedCrop(1000, scale=(0.5, 1)),
    # transforms.RandomHorizontalFlip(),
    # transforms.RandomVerticalFlip(),
    transforms.RandomRotation(
        degrees=90,
        expand=True,
        center=None,
        fill=None,
    ),

    transforms.ToTensor(),
    #     transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
    #
])

# transformer = transforms.Compose([
#     transforms.CenterCrop(1000),
#     transforms.PILToTensor(),
#     transforms.ConvertImageDtype(torch.float),
# ])
img = Image.open("./samples/baiduload/img_1.jpg")
test_a = transformer(img)
# test_a = transforms.ToTensor()(test_a)
print(res := test_a.numpy().T)
# # plt.imshow(res)
cv2.imshow('1', res)
cv2.waitKey(0)