MCPcopy Index your code
hub / github.com/facebookresearch/pytorchvideo

github.com/facebookresearch/pytorchvideo @0.1.3 sqlite

repository ↗ · DeepWiki ↗ · release 0.1.3 ↗
997 symbols 3,787 edges 157 files 552 documented · 55%
README

CircleCI CircleCI Coverage

<i> A deep learning library for video understanding research.</i>






<i>Check the <a href="https://pytorchvideo.org/">website</a> for more information.</i>
A PyTorchVideo-accelerated X3D model running on a Samsung Galaxy S10 phone. The model runs ~8x faster than real time, requiring roughly 130 ms to process one second of video. A PyTorchVideo-based SlowFast model performing video action detection.

Introduction

PyTorchVideo is a deeplearning library with a focus on video understanding work. PytorchVideo provides reusable, modular and efficient components needed to accelerate the video understanding research. PyTorchVideo is developed using PyTorch and supports different deeplearning video components like video models, video datasets, and video-specific transforms.

Key features include:

  • Based on PyTorch: Built using PyTorch. Makes it easy to use all of the PyTorch-ecosystem components.
  • Reproducible Model Zoo: Variety of state of the art pretrained video models and their associated benchmarks that are ready to use. Complementing the model zoo, PyTorchVideo comes with extensive data loaders supporting different datasets.
  • Efficient Video Components: Video-focused fast and efficient components that are easy to use. Supports accelerated inference on hardware.

Updates

Installation

Install PyTorchVideo inside a conda environment(Python >=3.7) with

pip install pytorchvideo

For detailed instructions please refer to INSTALL.md.

License

PyTorchVideo is released under the Apache 2.0 License.

Tutorials

Get started with PyTorchVideo by trying out one of our tutorials or by running examples in the tutorials folder.

Model Zoo and Baselines

We provide a large set of baseline results and trained models available for download in the PyTorchVideo Model Zoo.

Contributors

Here is the growing list of PyTorchVideo contributors in alphabetical order (let us know if you would like to be added): Aaron Adcock, Amy Bearman, Bernard Nguyen, Bo Xiong, Chengyuan Yan, Christoph Feichtenhofer, Dave Schnizlein, Haoqi Fan, Heng Wang, Jackson Hamburger, Jitendra Malik, Kalyan Vasudev Alwala, Matt Feiszli, Nikhila Ravi, Ross Girshick, Tullie Murrell, Wan-Yen Lo, Weiyao Wang, Yanghao Li, Yilei Li, Zhengxing Chen, Zhicheng Yan.

Development

We welcome new contributions to PyTorchVideo and we will be actively maintaining this library! Please refer to CONTRIBUTING.md for full instructions on how to run the code, tests and linter, and submit your pull requests.

Citing PyTorchVideo

If you find PyTorchVideo useful in your work, please use the following BibTeX entry for citation.

@inproceedings{fan2021pytorchvideo,
    author =       {Haoqi Fan and Tullie Murrell and Heng Wang and Kalyan Vasudev Alwala and Yanghao Li and Yilei Li and Bo Xiong and Nikhila Ravi and Meng Li and Haichuan Yang and  Jitendra Malik and Ross Girshick and Matt Feiszli and Aaron Adcock and Wan-Yen Lo and Christoph Feichtenhofer},
    title = {{PyTorchVideo}: A Deep Learning Library for Video Understanding},
    booktitle = {Proceedings of the 29th ACM International Conference on Multimedia},
    year = {2021},
    note = {\url{https://pytorchvideo.org/}},
}

Core symbols most depended-on inside this repo

thwc_to_cthw
called by 45
pytorchvideo/data/utils.py
create_dummy_video_frames
called by 42
tests/utils.py
close
called by 25
pytorchvideo/data/video.py
make_clip_sampler
called by 23
pytorchvideo/data/clip_sampling.py
set_attributes
called by 21
pytorchvideo/layers/utils.py
get_clip
called by 20
pytorchvideo/data/video.py
from_path
called by 19
pytorchvideo/data/encoded_video.py
save_dataclass_objs_to_headered_csv
called by 18
pytorchvideo/data/utils.py

Shape

Method 601
Function 200
Class 196

Languages

Python98%
TypeScript2%

Modules by API surface

pytorchvideo/transforms/transforms.py36 symbols
tests/test_transforms.py33 symbols
tests/test_models_resnet.py30 symbols
pytorchvideo/transforms/augmentations.py24 symbols
tests/test_data_labeled_video_dataset.py22 symbols
tutorials/video_classification_example/train.py21 symbols
pytorchvideo/models/masked_multistream.py21 symbols
pytorchvideo/layers/accelerator/mobile_cpu/convolutions.py20 symbols
pytorchvideo/models/resnet.py19 symbols
pytorchvideo/layers/accelerator/mobile_cpu/activation_functions.py19 symbols
tutorials/video_detection_example/visualization.py18 symbols
pytorchvideo/transforms/functional.py17 symbols

Dependencies from manifests, versioned

docusaurus1.14.6 · 1×
docutils0.16 · 1×
recommonmark0.6.0 · 1×
sphinx3.2.0 · 1×

For agents

$ claude mcp add pytorchvideo \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact