torch.torchvision
- 2019 年 10 月 7 日
- 筆記
版權聲明:本文為部落客原創文章,遵循 CC 4.0 BY-SA 版權協議,轉載請附上原文出處鏈接和本聲明。
本文鏈接:https://blog.csdn.net/weixin_36670529/article/details/101068646
torchvision包由流行的數據集、模型架構和用於電腦視覺的常見影像轉換組成。
Package Reference
- torchvision.datasets
- torchvision.io
- torchvision.models
- torchvision.ops
- torchvision.transforms
- torchvision.utils
torchvision.get_image_backend
()[source]
獲取用於載入影像的包的名稱
torchvision.set_image_backend
(backend)[source]
指定用於載入影像的包。
參數:
backend (string) – 映像後端名稱。其中一個{' PIL ', ' accimage '}。accimage包使用Intel IPP庫。它一般比PIL快,但是不支援那麼多的操作。