pytorch/vision

Health v2 · maintenance only

Adoption not part of the score

17885 stars · 7250 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

torchvision is the official PyTorch companion library providing datasets, model architectures, and image/video transformations for computer vision. It includes pretrained models and utilities that integrate tightly with PyTorch.

Use cases

• load standard vision datasets like CIFAR and ImageNet
• apply image transforms and augmentations for training
• use pretrained models like ResNet for image classification
• run object detection and segmentation models
• read and decode images and videos into tensors

When to choose

• you are training or evaluating computer vision models in PyTorch
• you need pretrained vision model architectures with weights
• you want standard dataset loaders and augmentation pipelines

When to avoid

• you use TensorFlow, JAX, or another non-PyTorch framework
• you need general-purpose image editing rather than ML pipelines

Facets

library · maturity stable

machine-learning image-processing computer-vision data-science computer-vision machine-learning deep-learning image-processing python cross-platform torchvision pytorch datasets transforms pretrained-models image-classification object-detection video gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="pytorch/vision")