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')