关于pytorch里面的图像变换

关于读入图片:

image = Image.open("F:\liwei\my_data\single_channel\BPSK\BPSK.1.jpg")

看一看图片的样式

print(image.size, image.format, image.mode)

裁剪图片:
裁剪成32像素的,短边resize成这个尺寸,长边安装对应比例进行缩放

resize=torchvision.transforms.Resize(32)
image=resize(image)

然后可以进行中心裁剪:

crop_obj = torchvision.transforms.CenterCrop(32)
image = crop_obj(image)

这时候裁剪完成就是一个32*32的图像了

我们可以把所有的图像转换放到一个compose里面

data_transform = transforms.Compose([
    transforms.Resize(32),
    transforms.CenterCrop(22),
    transforms.ToTensor(),# 转化为张量
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
    # 标准化
])

猜你喜欢

转载自blog.csdn.net/huatianxue/article/details/108320568