torchvisionΒΆ

The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.