导入 torchvision 库

从这篇文章开始,我们学习 PyTorch 在计算机视觉方面的应用。

我们之前使用的 torch 库,是深度学习框架,提供张量计算、自动梯度计算等功能。而本章刚接触的 torchvision 库,专注于计算机视觉,提供数据集和模型工具。

torchvision.datasets 提供常见数据集的加载方法。

torchvision.transforms 提供一系列图像预处理和数据增强工具。

如代码清单 1 所示,我们导入了 torch、torchvision 和 matplot 相关的库,并查看相关库版本进行验证。

代码清单 1 torchvision
  1. import torch
  2. from torch import nn
  3.  
  4. import torchvision
  5. from torchvision import datasets
  6. from torchvision import transforms
  7. from torchvision.transforms import ToTensor
  8.  
  9. import matplotlib.pyplot as plt
  10.  
  11. print(f"PyTorch 版本: {torch.__version__}")
  12. print(f"torchvision 版本: {torchvision.__version__}")