PyTorch torchvision

2020-09-15 14:07 更新

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

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号