torchvision 参考

译者：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要少。| | —- | —- |