Overview of Torchvision modules
The
torchvision library provides three main modules for computer vision tasks: datasets for common dataset wrappers, models for classic pre-defined network architectures, and transforms for image preprocessing operations.