【python深度学习】——torchvision.models

1. torchvision简介

torchvision是PyTorch生态系统中的一个包，专门用于计算机视觉任务。它提供了一系列用于加载、处理和预处理图像和视频数据的工具，以及常用的计算机视觉模型和数据集。

2. models模块

2.1 简介

关于此模块的官网介绍在这里。
这个模块包含许多常用的预训练计算机视觉模型，例如ResNet、AlexNet、VGG等分类、分割等模型。在官网示例中可以看到, 在0.14版本之后, 可以通过调用list_model函数来查看torch vision中提供的模型:

# List available models
all_models = list_models()
classification_models = list_models(module