参考 torch.torchvision - 云+社区 - 腾讯云
torchvision包由流行的数据集、模型架构和用于计算机视觉的常见图像转换组成。
Package Reference
- torchvision.datasets
- torchvision.io
- torchvision.models
- torchvision.ops
- torchvision.transforms
- torchvision.utils
torchvision.get_image_backend()[source]
获取用于加载图像的包的名称
torchvision.set_image_backend(backend)[source]
指定用于加载图像的包。
参数:
- backend (string) – 映像后端名称。其中一个{' PIL ', ' accimage '}。accimage包使用Intel IPP库。它一般比PIL快,但是不支持那么多的操作。
本文介绍了torchvision库的主要组成部分,包括流行的数据集、模型架构及图像转换等。详细列举了如MNIST、CIFAR等数据集,ResNet等模型以及各种图像转换操作。
13万+

被折叠的 条评论
为什么被折叠?



