torch:)-detailed explanation of torchvision pytorch graphics library


torchvision It is an image library of pytorch that serves the pytorch deep learning framework and is mainly used to build computer vision models.

  1. torchvision.datasets: Some functions for loading data and excuses for commonly used data sets
  2. torchvision.models: Contains commonly used model structures (including pre-training models), such as AlexNet, VGG, ResNet, etc.
  3. torchvision.transforms: Common image transformations, such as cropping, rotating, etc.
  4. torchvision.utils: Some other useful methods

1. Detailed usage of torchvision.transforms

Click to view: detailed usage of torchvision.transforms

Guess you like

Origin blog.csdn.net/fuhao7i/article/details/109645294