Plain-AyI
← All projects
Developer building blocks

torchvision

Ready-made image datasets, models and photo tools for building computer-vision AI with PyTorch.

What it does

torchvision is a companion library to PyTorch for computer vision, the area of AI that works with images. It bundles popular image datasets, well-known model designs and common image adjustments like resizing and cropping.

Its makers note that it only downloads public datasets and doesn't host them, and that users must check they have permission to use each dataset and pre-trained model.

Who it's for

Programmers and researchers building AI that works with images.

Can I use it without coding?

Developers only. It's a code library installed alongside PyTorch. New to this? Start here.

Can I run it?

Cost
FreeNothing to pay or sign up for.
API key
Not needed

Try it

Using Claude Code, Cursor or another AI coding assistant? Paste the prompt and it will walk you through the setup.New to this?

Before you install

  • Only download from the official GitHub page. Copies with similar names exist for many popular projects.
  • License (BSD-3-Clause): Permissive: free to use, change and share, including for business.
  • More on staying safe

Where people are talking about it