torchvision¶
此庫是 PyTorch 專案的一部分。PyTorch 是一個開源機器學習框架。
本 文件中描述的功能按釋出狀態分類
穩定版:這些功能將得到長期維護,通常不會有重大的效能限制或文件缺失。我們也期望保持向後相容性(儘管可能發生重大更改,並會在提前一個版本時發出通知)。
Beta 版:功能被標記為 Beta 版是因為 API 可能根據使用者反饋而更改,或者是因為效能需要改進,或者是因為運算子的覆蓋範圍尚未完成。對於 Beta 版功能,我們致力於將其推進到穩定版分類。但是,我們不承諾向後相容性。
原型:這些功能通常不作為 PyPI 或 Conda 等二進位制發行版的一部分提供,除非有時透過執行時標誌,並且處於反饋和測試的早期階段。
該 torchvision 包包含用於計算機視覺的流行資料集、模型架構和常用影像轉換。
- torchvision.get_video_backend()[原始碼]¶
返回當前用於解碼影片的影片後端。
- 返回:
影片後端的名稱。是 {‘pyav’, ‘video_reader’} 之一。
- 返回型別:
- torchvision.set_image_backend(backend)[原始碼]¶
指定用於載入影像的包。
- 引數:
backend (string) – 影像後端的名稱。是 {‘PIL’, ‘accimage’} 之一。
accimage包使用 Intel IPP 庫。它通常比 PIL 快,但支援的操作不如 PIL 多。
- torchvision.set_video_backend(backend)[原始碼]¶
指定用於解碼影片的包。
- 引數:
backend (string) – 影片後端的名稱。是 {‘pyav’, ‘video_reader’} 之一。
pyav包使用第三方 PyAv 庫。它是 FFmpeg 庫的 Pythonic 繫結。video_reader包包含基於 FFMPEG 庫的原生 C++ 實現,以及 TorchScript 自定義運算元的 Python API。它通常比pyav解碼速度更快,但可能不如pyav健壯。
注意
在最新的 main 中,預設停用了 FFMPEG 的構建。如果您想使用 ‘video_reader’ 後端,請從原始碼編譯 torchvision。