【plt顯示Tensor轉出來的array時的報錯】TypeError: Invalid dimensions for image data

簡述

當使用plt時去顯示Tensor轉出來的np.array時候,遇到這個問題。

這是因爲你使用的圖是彩圖(三維)導致的。

且由於plt顯示的時候,對於三維的圖片要求的維度是[長,寬,通道數]

解決方案

對於數組處理下對應的維度即:

array = array.transpose((1, 2, 0)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章