← all repositories

pytorch/vision

Official PyTorch computer vision library providing datasets, pre-trained models, and image transformation utilities.

17.7k stars Python Computer VisionML Frameworks
vision
Velocity · 7d
+5.1
★ / day
Trend
steady
star history

torchvision is the official PyTorch computer vision library that provides standard benchmark datasets (CIFAR, ImageNet, etc.), common image transformations (resizing, cropping, normalization), and pre-trained model architectures (ResNet, VGG, etc.) used for building and training deep learning vision systems. It integrates directly with PyTorch’s tensor and autograd machinery for seamless training pipelines.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.