在pytorch中圖片的張量結(jié)構(gòu)與plt可以顯示的圖片格式要求是不一樣的,所以plt是不能直接顯示tensor格式的圖片的,那么pytorch怎么用plt顯示tensor圖片呢?這就需要涉及到數(shù)據(jù)轉(zhuǎn)換了,基本思路就是將tensor轉(zhuǎn)換為numpy類型的數(shù)據(jù)結(jié)構(gòu),而numpy類型的格式剛好可以被plt支持。接下來就來看具體怎么操作吧!
問題
圖像的張量結(jié)構(gòu)為(C,H,W),而plt可以顯示的圖片格式要求(H,W,C),C為顏色通道數(shù),可以沒有。
所以問題就是將Tensor(C,H,W)=> numpy(H,W,C)
解決辦法
def transimg(img):
img = img / 2 + 0.5 # unnormalize
npimg = img.numpy()
npimg1 = np.transpose(npimg,(1,2,0)) # C*H*W => H*W*C
return npimg1
以上就是pytorch怎么用plt顯示tensor的方法介紹了,希望能給大家一個參考,也希望大家多多支持W3Cschool。