Skip to content

torchvision

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

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

包装参考


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 快,但可能不够鲁棒。



回到顶部