导入 torchvision 库
从这篇文章开始,我们学习 PyTorch 在计算机视觉方面的应用。
我们之前使用的 torch 库,是深度学习框架,提供张量计算、自动梯度计算等功能。而本章刚接触的 torchvision 库,专注于计算机视觉,提供数据集和模型工具。
torchvision.datasets 提供常见数据集的加载方法。
torchvision.transforms 提供一系列图像预处理和数据增强工具。
如代码清单 1 所示,我们导入了 torch、torchvision 和 matplot 相关的库,并查看相关库版本进行验证。
- import torch
- from torch import nn
- import torchvision
- from torchvision import datasets
- from torchvision import transforms
- from torchvision.transforms import ToTensor
- import matplotlib.pyplot as plt
- print(f"PyTorch 版本: {torch.__version__}")
- print(f"torchvision 版本: {torchvision.__version__}")