torchvision是一個包,它服務于pytorch深度學習架構,用來生成圖檔,視訊資料集,模型類和預訓練的模型
torchvision由以下四個部分組成:
1. torchvision.datasets : Data loaders for popular vision datasets
2. torchvision.models : Definitions for popular model architectures, such as AlexNet, VGG, and ResNet and pre-trained models.
3. torchvision.transforms : Common image transformations such as random crop, rotations etc.
4. torchvision.utils : Useful stuff such as saving tensor (3 x H x W) as image to disk, given a mini-batch creating a grid of images, etc.
個人學習記錄