pytorch中怎么用plt顯示tensor？

在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支持。接下來(lái)就來(lái)看具體怎么操作吧！

問題

圖像的張量結(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的方法介紹了，希望能給大家一個(gè)參考，也希望大家多多支持W3Cschool。