PyTorch torchvision

torchvision

原文： PyTorch torchvision參考

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

torchvision.get_image_backend()?

獲取用于加載圖像的包的名稱(chēng)

torchvision.set_image_backend(backend)?

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

參數(shù)

后端(字符串）–圖像后端的名稱(chēng)。 {'PIL'，'accimage'}之一。 accimage程序包使用 Intel IPP 庫(kù)。 它通常比 PIL 快，但不支持那么多操作。

torchvision.set_video_backend(backend)?

指定用于解碼視頻的包。

Parameters

后端(字符串）–

視頻后端的名稱(chēng)。 {'pyav'，'video_reader'}之一。 pyav包使用第三方 PyAv 庫(kù)。 這是 Pythonic

FFmpeg 庫(kù)的綁定。

The video_reader package includes a native c++ implementation on

FFMPEG 庫(kù)的頂部，以及 TorchScript 自定義運(yùn)算符的 python API。 通常，它的解碼速度比 pyav 快，但可能不夠魯棒。