PyTorch torchvision

torchvision

原文： PyTorch torchvision參考

torchvision 軟件包包含用于計算機視覺的流行數(shù)據(jù)集，模型架構(gòu)和常見圖像轉(zhuǎn)換。

torchvision.get_image_backend()?

獲取用于加載圖像的包的名稱

torchvision.set_image_backend(backend)?

指定用于加載圖像的軟件包。

參數(shù)

后端(字符串）–圖像后端的名稱。 {'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 快，但可能不夠魯棒。