PyTorch torchvision

torchvision

原文： PyTorch torchvision参考

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