AI Potluck
Back to Gap Map Infrastructure / Classic ML & computer vision

torchvision

PyTorch Foundation (Linux Foundation)
open source / Overall score: 4.2(strong)

PyTorch's computer-vision library, with popular datasets, image and video transforms, vision operators, and model architectures with pretrained weights for classification, semantic and instance segmentation, object detection, keypoint detection, video classification and optical flow. Reference training scripts reproduce each model. It is part of the PyTorch project under the PyTorch Foundation.

Openness

5 high confidence
5.0
license
BSD-3-Clause(OSI
source
public(github.com/pytorch/vision)
core features withheld
no — PyTorch Foundation project

The library is BSD-3-Clause and builds from the public repository. Its pretrained weights can carry their own terms from the data they were trained on, and the SWAG weights are non-commercial; they are optional downloads, not the library.

Adoption

5 medium confidence
5.0

Measured on monthly PyPI downloads of torchvision. Many of them come from installing it alongside PyTorch rather than from choosing it, so the figure overstates deliberate use.

Capability

3 high confidence
3.0

torchvision is a model zoo with the datasets, transforms and training recipes to fine-tune it, like timm, whose catalog of image backbones is larger. Its pretrained models predict the classes of the benchmarks they were trained on, so a new task still means a training run.

  • https://docs.pytorch.org/vision/stable/models.html recorded 2026-09-27

    Models page: "TorchVision offers pre-trained weights for every provided architecture"; sections for classification, quantized models, semantic segmentation, object detection, instance segmentation, keypoint detection, video classification and optical flow.

Verified 2026-09-27