facebookresearch/pytorchvideo

A deep learning library for video understanding research.

PyTorchVideo – ビデオ理解のためのディープラーニングライブラリ

概要 – PyTorchVideo は、PyTorch 上に構築されたオープンソースの Python ライブラリで、ビデオベースの AI モデルの研究と開発のための既製コンポーネントを提供します。ビデオ特有のデータローダー、変換、最先端アーキテクチャのモデル zoo(例:SlowFast、X3D、Multiscale Vision Transformers)に加え、効率的なトレーニングと推論のためのユーティリティをまとめています。

重要性 – ビデオは視覚データに時間的次元を追加するため、静止画像よりもはるかに要求が高くなります。PyTorchVideo は、フレーム抽出、時間的拡張、バッチ処理、GPU アクセラレーション操作などの低レベルの配管を抽象化し、研究者がエンジニアリングの詳細ではなくモデルのアイデアに集中できるようにします。また、事前学習済みチェックポイントとベンチマーク結果を提供し、再現可能な実験を可能にします。

主な特徴

  • PyTorch ネイティブ – より広範な PyTorch エコシステム(オプティマイザー、スケジューラー、分散トレーニング、TorchVision など)とシームレスに連携します。
  • モデル zoo – 数十の事前学習済みビデオモデル(SlowFast、X3D、MViT など)とベンチマークスコア、ワンクリックダウンロードスクリプトを提供。
  • 効率的なビデオパイプライン – 高速なビデオ特有の変換、時間サンプリング、CPU、GPU、モバイルデバイスでの高速推論をサポート(例:X3D モデルは Galaxy S10 でリアルタイムの約 8 倍の速度で動作)。
  • データセットローダー – 一般的なビデオデータセット(Kinetics、AVA、Something-Something など)の組み込みサポートと柔軟な設定。
  • 拡張性 – モジュール設計により、コアコードを書き換えることなく、カスタムモデル、変換、データソースをプラグインできます。
  • ドキュメントとチュートリアル – クイックスタートガイド、サンプルノートブック、リポジトリ内の「tutorials」フォルダを備えたウェブサイト。

典型的なユースケース

  • アクション認識、ビデオ分類、時空間検出に関する学術研究。
  • ビデオ中心の AI 製品のプロトタイピング(例:リアルタイムアクティビティ監視、ビデオ検索、スポーツ分析)。
  • 新しいビデオアーキテクチャを既存のベースラインとベンチマークする。
  • エッジデバイスへの軽量ビデオモデルの展開(リポジトリはモバイル向け X3D デモを紹介)。

はじめに

# Python ≥3.7 の conda 環境内で
pip install pytorchvideo

インストール後、公式サイトのチュートリアルに従うか、tutorials/ ディレクトリの例を実行してください。

コミュニティとサポート

  • Slack: README のバッジから #pytorchvideo チャンネルに参加。
  • GitHub: イシュートラッカー、プルリクエストワークフロー、貢献ガイドライン(CONTRIBUTING.md)。
  • モデル zoo: docs/source/model_zoo.md に文書化され、事前学習済み重みへのリンクがあります。

ライセンス – Apache 2.0。自由な使用、変更、配布が許可されています。

引用 – 出版物で PyTorchVideo を使用する場合は、付随する ACM MM 2021 論文(BibTeX は README に記載)を引用してください。


上記の情報はすべて、リポジトリの README およびリンクされたドキュメントから直接取得しています。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト