参考   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快，但是不支持那么多的操作。