pycharm調試模式中如何將ndarray二維矩陣變成圖片查看

pycharm調試模式中如何將ndarray二維矩陣變成圖片查看

1、調試窗口中，右鍵，選擇Evaluate Expression。
2、輸入下面的代碼，其中transpose操作是爲了轉換圖片的shape（pytorch中需要，其他不需要），反正只要把矩陣轉換成HxWxC的樣式就行了，比方說可以是500x500x500，而不能是3x500x500
我這裏需要查看的是t[0]這個變量：

import matplotlib.pyplot as plt
pic = np.transpose(t[0], axes=[2, 1, 0])
plt.imshow(pic)
plt.show()

輸入如下

點擊Evaluate之後，右側可以顯示出該圖片：

更加便捷的方法，定義好輸入模板

設置中增加一個live template：

裏面填寫的內容:

import matplotlib.pyplot as plt
debug_show_pic = $ARRAY_NAME$
# 下面這行是爲了轉換shape,不需要可以刪掉
debug_show_pic = np.transpose(debug_show_pic, axes=[2, 1, 0])
plt.imshow(debug_show_pic)
plt.show()

然後debug的時候，輸入debug_show就行了。