PyTorch 1.0 中文文档:torchvision 参考

2022-05-07 14:24:16 浏览数 (2)

译者: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

阅读全文/改进本文

0 人点赞