torchvision
It is an image library of pytorch that serves the pytorch deep learning framework and is mainly used to build computer vision models.
torchvision.datasets
: Some functions for loading data and excuses for commonly used data sets
torchvision.models
: Contains commonly used model structures (including pre-training models), such as AlexNet, VGG, ResNet, etc.
torchvision.transforms
: Common image transformations, such as cropping, rotating, etc.
torchvision.utils
: Some other useful methods
1. Detailed usage of torchvision.transforms
Click to view: detailed usage of torchvision.transforms