資料結構
網路
關係資料庫管理系統 (RDBMS)
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理學
化學
生物學
數學
英語
經濟學
心理學
社會學
服裝設計
法律研究如何將Torch張量轉換為PIL影像？
ToPILImage() 變換將 torch 張量轉換為 PIL 影像。**torchvision.transforms** 模組提供了許多重要的變換，可用於對影像資料執行不同型別的操作。**ToPILImage()** 接受形狀為 **[C, H, W]** 的 torch 張量，其中 **C、H** 和 **W** 分別是相應 PIL 影像的通道數、影像高度和寬度。
步驟
我們可以使用以下步驟將 torch 張量轉換為 PIL 影像：
匯入所需的庫。在以下所有示例中，所需的 Python 庫為 **torch、Pillow** 和 **torchvision**。確保您已安裝它們。
import torch import torchvision import torchvision.transforms as T from PIL import Image
定義形狀為 **[C, H, W]** 的 torch 張量。
tensor = torch.rand(3,256,256)
定義一個變換，將 torch 張量轉換為 PIL 影像。
transform = T.ToPILImage()
將上述定義的變換應用於輸入 torch 張量，將其轉換為 PIL 影像。
img = transform(tensor)
顯示轉換後的 PIL 影像。
img.show()
示例
請看下面的示例：
# import required libraries import torch import torchvision.transforms as T from PIL import Image # define a torch tensor tensor = torch.rand(3,300,700) # define a transform to convert a tensor to PIL image transform = T.ToPILImage() # convert the tensor to PIL image using above transform img = transform(tensor) # display the PIL image img.show()
輸出
它將產生以下輸出：
更新於：2023年9月10日
42K+ 次瀏覽
廣告