什么是torchvision

该torchvision软件包包括流行的数据集,模型体系结构和用于计算机视觉的常见图像转换。

https://pytorch.apachecn.org/docs/1.2/torchvision/

https://pytorch.org/docs/stable/torchvision/index.html

 

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

你可能感兴趣的:(深度学习)