pytorch/vision

Datasets, Transforms and Models specific to Computer Vision

What it solves

它提供了一套标准化的计算机视觉任务工具，减少了开发者从头开始手动实现常见数据集、模型架构和图像处理步骤的需求。

How it works

Torchvision 作为 PyTorch 的工具库，，提供了一系列可供下载和准备的热门数据集、预定义模型架构和常见图像转换。它支持多种图像后端，包括 torch tensors 和 PIL images (通过 Pillow 或 Pillow-SIMD)。

Who it's for

针对使用 PyTorch 生态系统构建计算机视觉应用的研究人员和开发者。

Highlights

  • 热门计算机视觉数据集
  • 预定义模型架构
  • 常见图像转换
  • 支持多种图像后端，例如 Pillow 和 Pillow-SIMD