Torch中的图像处理模块主要包括以下几个:
torchvision.transforms:用于图像的常见变换和预处理操作,如缩放、裁剪、旋转、翻转等。
torch.nn.functional:包含一些图像处理相关的函数,如卷积操作、池化操作、激活函数等。
torchvision.datasets:包含一些常用的图像数据集,如MNIST、CIFAR等。
torch.utils.data.DataLoader:用于加载和处理图像数据,支持批量加载、数据增强等功能。
torch.optim:包含各种优化算法,用于训练图像处理模型。
torchvision.models:包含一些常用的图像处理模型,如ResNet、VGG等。
torchvision.utils:包含一些辅助函数,用于可视化图像数据。