torchvision 0.13.1


image and video datasets and models for torch deep learning

image and video datasets and models for torch deep learning

Stars: 12440, Watchers: 12440, Forks: 6223, Open Issues: 776

The pytorch/vision repo was created 5 years ago and was last updated an hour ago.
The project is extremely popular with a mindblowing 12440 github stars!

How to Install torchvision

You can install torchvision using pip

pip install torchvision

or add it to a project with poetry

poetry add torchvision

Package Details

PyTorch Core Team
GitHub Repo
No  torchvision  pypi packages just yet.


A list of common torchvision errors.

Code Examples

Here are some torchvision code examples and snippets.

GitHub Issues

The torchvision package has 776 open issues on GitHub

  • fix bug when the target is empty
  • FCOS empty box images
  • Replace Assert statement with ValueError message.
  • changed assert to ValueError in backbone_utils
  • Add support for get_metadata() in GPU decoder
  • [WIP] Adding more ConvNeXt variants
  • video classification experiments using GPU decoder
  • video classification reference script with GPU decoder support
  • Add kinetics dataset to use new video reading API
  • Add support for get_metadata() in GPU decoder
  • Vectorize box decoding in FCOS model
  • [PoC] add test to check available feature types of prototype datasets
  • Change CPU decoder output frames to use ITU709 colour space
  • Allow more consistent degrees=None in RandomAffine constructor
  • [feature request] Derive all transforms classes from nn.Module?

See more issues on GitHub

Related Packages & Articles

mediapipe 0.8.11

MediaPipe is the simplest way for researchers and developers to build world-class ML solutions and applications for mobile, edge, cloud and the web.

kornia 0.6.7

Open Source Differentiable Computer Vision Library for PyTorch

dlib 19.24.0

A toolkit for making real world machine learning and data analysis applications

datasets 2.5.0

HuggingFace community-driven open-source library of datasets

nlp 0.4.0

HuggingFace/NLP is an open library of NLP datasets.

pytorch-lightning 1.7.6

PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.

pyAudioAnalysis 0.3.14

Python Audio Analysis Library: Feature Extraction, Classification, Segmentation and Applications