关于读入图片:
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])
# 标准化
])