torchvision 参考
优质
小牛编辑
133浏览
2023-12-01
译者：BXuan694
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
torchvision.get_image_backend()
查看载入图片的包的名称
torchvision.set_image_backend(backend)
指定用于载入图片的包
| 参数: | backend (string) – 图片处理后端的名称，须为{‘PIL’, ‘accimage’}中的一个。accimage包使用了英特尔IPP库。这个库通常比PIL快，但是支持的操作比PIL要少。 |
|---|