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 などの複数の画像バックエンドのサポート