使用matplotlib.pyplot.imshow() 显示图像时出现“TypeError: Invalid dimensions for image data”的问题
转载自：https://blog.****.net/qq_17478877/article/details/88977295，本文只做个人记录学习使用，版权归原作者所有。
如上两幅图，matplotlib.pyplot.imshow()需要数据是二维的数组或者第三维深度是3或4的三维数组，当第三维深度为1时，使用np.squeeze()压缩数据成为二维数组