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就行了。