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 およびリンクされたドキュメントから直接取得したものです。

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트