【PyTorch】站在PyTorch的肩膀上玩CV：TorchVision

Backto PyTorch Index

TorchVision 是基於 PyTorch 深度定製的一個 CV 包，包含 CVer 常用的 數據集，模型，圖像操作等。注意，TorchVison 一直在更新中，代碼結構變化也很大，一定要注意安裝與 PyTorch, Python 相匹配的版本。最新支持的數據集、模型、操作等，參看最新的 文檔.

Tips

set image backend

torchvison 默認使用 PIL 作爲圖像處理 backend，建議切換爲 accimage, Intel CPU下有加速，處理快。The accimage package uses the Intel IPP library. It is generally faster than PIL, but does not support as many operations.

torchvision.set_image_backend('accimage')