torch.torchvision

  • 2019 年 10 月 7 日
  • 筆記

版權聲明:本文為部落客原創文章,遵循 CC 4.0 BY-SA 版權協議,轉載請附上原文出處鏈接和本聲明。

本文鏈接:https://blog.csdn.net/weixin_36670529/article/details/101068646

torchvision包由流行的數據集、模型架構和用於電腦視覺的常見影像轉換組成。

Package Reference

torchvision.get_image_backend()[source]

獲取用於載入影像的包的名稱

torchvision.set_image_backend(backend)[source]

指定用於載入影像的包。

參數:

backend (string) – 映像後端名稱。其中一個{' PIL ', ' accimage '}。accimage包使用Intel IPP庫。它一般比PIL快,但是不支援那麼多的操作。