pytorch学习（4）——图片变换Transforms使用
1 Transforms类的使用
为图像变换的函数库。
1.1 导入transforms库
python脚本内输入：
from torchvision import transforms
按住Ctrl，点击transforms。
进入__init__.py。
from .transforms import *
from .autoaugment import *
按住Ctrl，点击.transforms。
进入transforms.py。
1.2 Transforms结构及用法
transforms.py工具箱内的函数主要对图像进行变换，包括数据结构转换、尺度变换、高斯模糊、归一化、流水线、裁剪等。
点击Structure（ALT+7）查看包含的函数名。
- Compose：流水线操作，将多个图像操作步骤整合到一起。
- ToTensor：将PIL或ndarray类型的图像转换成张量（tensor）类型。
- Resize：将输入的图像转换成不同的大小。
- CenterCrop：对输入的图像进行中心裁剪。
- ToPILImage：将tensor或ndarray类型的图像转换成PIL类型的图像。
- GaussianBlur：对输入的图像进行高斯模糊处理。
大多数函数的工作流程：
Compose函数的工作流程：
代码如下：
from PIL import Image
from torchvision import transforms
img_path_abs = "G:\\ant.jpg"
img = Image.open(img_path_abs)
data_transform = transforms.Compose([
transforms.Resize((256, 256)), # 调整图像大小为256x256
transforms.RandomCrop((224, 224)), # 随机裁剪图像为224x224
transforms.ToTensor(), # 将图像转换为张量
transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5]) # 标准化图像张量
])
transformed_data = data_transform(image)
python脚本代码如下：
from PIL import Image
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
img_path_abs = "G:\\ant.jpg"
img = Image.open(img_path_abs)
print(img)
# 如果取消下方注释，则会在tensorboard中写入图像
# writer = SummaryWriter("logs")
# transforms.ToTensor PIL -> tensor
trans_totensor = transforms.ToTensor() # 创建具体工具，实例化对象
tensor_img = trans_totensor(img)
# 以上两步也可以简化为：
# tensor_img = transforms.ToTensor()(img)
print(tensor_img)
try:
writer.add_image("ToTensor_img", tensor_img, 1)
except NameError:
print("writer未定义")
# transforms.ToPILImage tensor -> PIL
trans_PILImage = transforms.ToPILImage()
pil_img = trans_PILImage(tensor_img)
pil_img.show()
# transforms.Normalize 归一化
trans_norm = transforms.Normalize([1, 1, 2], [1, 2, 1])
img_norm = trans_norm(tensor_img)
print("tensor_img")
print(tensor_img[0][0][0])
print("\r\nimg_norm")
print(img_norm[0][0][0])
try:
writer.add_image("Normalize_img", img_norm, 1)
except NameError:
print("writer未定义")
# transforms.Resize -1 改变图像比例
print("img.size:" + str(img.size))
trans_resize = transforms.Resize((512, 512))
img_resize = trans_resize(img) # PIL(原图尺寸) -> PIL（512,512）
print("img_resize.size:" + str(img_resize.size))
print("img_resize:" + str(img_resize))
img.show()
img_resize.show()
try:
writer.add_image("Resize_img", img_resize, 1)
except NameError:
print("writer未定义")
# Resize -2 不改变图像比例
trans_resize_2 = transforms.Resize(1024) # PIL(原图尺寸) -> PIL（1024,*）
img_resize_2 = trans_resize_2(img)
print("img.size:" + str(img.size))
print("img_resize_2.size:" + str(img_resize_2.size))
img.show()
img_resize_2.show()
try:
writer.add_image("Resize_img_2", img_resize_2, 1)
except NameError:
print("writer未定义")
# RandomCrop
trans_randomcrop = transforms.RandomCrop(512)
img_randomcrop = trans_randomcrop(img)
img_randomcrop.show()
# Compose 流水线操作，将多个图像变换函数集成到一起，串联起来
trans_blur = transforms.GaussianBlur(kernel_size=5, sigma=(10, 10)) # 高斯模糊函数
trans_compose = transforms.Compose([trans_resize_2, trans_randomcrop, trans_totensor, trans_blur, trans_PILImage])
# 流水线操作，将多个变换串联在一起：等比缩放，PIL转tensor，高斯模糊，tensor转PIL
img_blur = trans_compose(img)
img_blur.show()
try:
writer.close()
except NameError:
print("writer未定义，无需关闭")