原文:https://pytorch.org/docs/stable/torchvision/index.html
torchvision
包收錄了若幹重要的公開資料集、網絡模型和計算機視覺中的常用圖像變換
包參考
- torchvision.datasets
- MNIST
- Fashion-MNIST
- EMNIST
- COCO
- LSUN
- ImageFolder
- DatasetFolder
- Imagenet-12
- CIFAR
- STL10
- SVHN
- PhotoTour
- SBU
- Flickr
- VOC
- torchvision.models
- Alexnet
- VGG
- ResNet
- SqueezeNet
- DenseNet
- Inception v3
- torchvision.transforms
- Transforms on PIL Image
- Transforms on torch.*Tensor
- Conversion Transforms
- Generic Transforms
- Functional Transforms
- torchvision.utils
檢視載入圖檔的包的名稱
指定用于載入圖檔的包
- 參數:backend(string)– 圖檔處理後端的名稱,須為
,PIL
中的一個。accimage
包使用了英特爾IPP庫。這個庫通常比PIL快,但是支援的操作比PIL要少。accimage