TorchVision官方文档翻译为中文-目录及相关图像视频解码包定义函数-001

PyTorch torchvision 模块详解：数据集、模型和图像处理工具

最新推荐文章于 2025-04-28 20:00:00 发布

原创

最新推荐文章于 2025-04-28 20:00:00 发布 · 655 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #神经网络

torchvision软件包包括流行的数据集、模型架构和用于计算机视觉的通用图像转换。
包参考
torchvision.datasets

Caltech
CelebA
CIFAR
Cityscapes
COCO
EMNIST
FakeData
Fashion-MNIST
Flickr
HMDB51
ImageNet
Kinetics-400
KITTI
KMNIST
LSUN
MNIST
Omniglot
PhotoTour
Places365
QMNIST
SBD
SBU
SEMEION
STL10
SVHN
UCF101
USPS
VOC
WIDERFace
Base classes for custom datasets

torchvision.io

Video
Fine-grained video API
Image

torchvision.models

Classification
Semantic Segmentation
Object Detection, Instance Segmentation and Person Keypoint Detection
Video classification

torchvision.ops
torchvision.transforms

Scriptable transforms
Compositions of transforms
Transforms on PIL Image and torch.*Tensor
Transforms on PIL Image only
Transforms on torch.*Tensor only
Conversion Transforms
Generic Transforms
AutoAugment Trans