1. 程式人生 > >pytorch載入和顯示圖片

pytorch載入和顯示圖片

程式碼實現如下:

import numpy as np
from torchvision import transforms
import cv2

#transforms.ToTensor()
transform1=transforms.Compose([transforms.ToTensor()])
#讀入影象
img=cv2.imread('cat.jpg')
img1=transform1(img) #歸一化到【0.0,1.0】
print ("img1=",img1)   #輸出的是畫素值,應該是Tensor

#轉化為numpy.ndarray並顯示
img_1=img1.numpy()*255
img_1=img_1.astype('uint8')
img_1=np.transpose(img_1,(1,2,0)) #將通道數移到最後
cv2.imshow('img_1',img_1)
cv2.waitKey()