torchvision
torchvision
软件包包含用于计算机视觉的流行数据集,模型架构和常见图像转换。
包装参考
- torchvision.datasets
- torchvision.io
- torchvision.models
- torchvision.ops
- torchvision.transforms
- torchvision.utils
torchvision.get_image_backend()¶
获取用于加载图像的包的名称
torchvision.set_image_backend(backend)¶
指定用于加载图像的软件包。
参数
后端(字符串)–图像后端的名称。 {'PIL','accimage'}之一。 accimage
程序包使用 Intel IPP 库。 它通常比 PIL 快,但不支持那么多操作。
torchvision.set_video_backend(backend)¶
指定用于解码视频的包。
Parameters
后端(字符串)–
视频后端的名称。 {'pyav','video_reader'}之一。 pyav
包使用第三方 PyAv 库。 这是 Pythonic
FFmpeg 库的绑定。
The video_reader package includes a native c++ implementation on
FFMPEG 库的顶部,以及 TorchScript 自定义运算符的 python API。 通常,它的解码速度比 pyav 快,但可能不够鲁棒。