小土堆pytorch学习笔记 之 transform 的使用

pytorch 之 transform 的使用

在 PyTorch 中，transforms 的作用包括：

1. 数据增强：通过随机变换增加数据集的多样性，比如随机旋转、翻转、缩放等，有助于提高模型的泛化能力。
2. 标准化：将数据转换成统一的格式或范围，比如将图像的像素值从 0-255 归一化到 0-1 之间（ToTensor）。
3. 数据预处理：在训练模型之前对数据进行必要的处理，比如中心剪裁、大小调整等，Resize

这里就以 ToTensor 为例介绍如何使用

Transforms.ToTensor()

transforms.ToTensor() 是一个常用的图像预处理操作，它将 PIL 图像或 NumPy 数组转换为 PyTorch 张量，并自动将图像的像素值从[0, 255]缩放到[0.0, 1.0]。

ToTensor 的使用

这是 transforms.py 中的定义，如何使用呢？

class ToTensor: