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