python深度学习通过torchaudio和torchvision

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

• 1. torchvision概述
  • 2. models组件
    • 2.1 简要说明
      • 2.1.1 模型导入——未包含预训练参数
      • 2.1.2 模型导入——设定特定预训练参数版本
      • 2.1.3 模型导入——调用量化处理后的模型版本

1. torchvision简介

作为PyTorch框架体系中的关键组件，torchvision包被设计用于支持各类计算机视觉相关应用。该模块集成了多种图像与视频数据的加载、处理及预处理功能，并整合了广泛使用的视觉模型与对应的数据集资源。

models模块功能与实现

2.1 简介

有关该模块的官方说明文档可参见这里。
此模块集成了大量广泛使用的预训练计算机视觉模型，包括但不限于ResNet、AlexNet、VGG等适用于分类与分割任务的模型。根据官网提供的示例信息，在0.14版本更新后，用户可通过调用list_model函数来获取torch vision中所支持的模型列表：

    # List available models
    all