torch.torchvision
参考 torch.torchvision - 云+社区 - 腾讯云
torchvision包由流行的数据集、模型架构和用于计算机视觉的常见图像转换组成。
Package Reference
- torchvision.datasets
- MNIST
- Fashion-MNIST
- KMNIST
- EMNIST
- QMNIST
- FakeData
- COCO
- LSUN
- ImageFolder
- DatasetFolder
- ImageNet
- CIFAR
- STL10
- SVHN
- PhotoTour
- SBU
- Flickr
- VOC
- Cityscapes
- SBD
- USPS
- Kinetics-400
- HMDB51
- UCF101
- torchvision.io
- Video
- torchvision.models
- Classification
- Semantic Segmentation
- Object Detection, Instance Segmentation and Person Keypoint Detection
- Video classification
- torchvision.ops
- torchvision.transforms
- Transforms on PIL Image
- Transforms on torch.*Tensor
- Conversion Transforms
- Generic Transforms
- Functional Transforms
- torchvision.utils
torchvision.get_image_backend()[source]
获取用于加载图像的包的名称
torchvision.set_image_backend(backend)[source]
指定用于加载图像的包。
参数:
- backend (string) – 映像后端名称。其中一个{' PIL '， ' accimage '}。accimage包使用Intel IPP库。它一般比PIL快，但是不支持那么多的操作。