Compose(transforms):将多个数据预处理函数组合成一个管道。
ToTensor():将 PIL 图像或 Numpy 数组转换为 PyTorch 张量。
Normalize(mean, std):对张量进行归一化,减去均值并除以标准差。
Resize(size):将图像缩放到指定大小。
CenterCrop(size):对图像进行中心裁剪。
RandomCrop(size):对图像进行随机裁剪。
RandomHorizontalFlip():以一定的概率对图像进行随机水平翻转。
RandomRotation(degrees):对图像进行随机旋转。
transforms 是 PyTorch 中 torchvision 库的一个模块,用于进行图像数据预处理和增强。它提供了一系列常用的数据预处理函数,如图像裁剪、缩放、翻转、归一化等。
transforms 模块中的函数通常用于创建数据预处理管道,以将输入数据转换为模型所需的格式。例如,在训练神经网络时,通常会对输入图像进行随机翻转、随机裁剪等增强操作,以扩展数据集并提高模型的泛化能力