使用pylab保存热图

Question

我正在使用ubuntu 14.04服务器并从红外摄像头捕获一些深度数据。

depth = get_depth()
print np.shape(depth)

输出为（480,640）。基本上它包含（x，y）坐标形式的深度值。

我想将这些数据保存为jpeg图片中的热图，然后从中创建一个MJPEG视频。但是，当我尝试

时

import pylab as pl
depth = get_depth()
pl.pcolor(data)  

它给了我

_tkinter.TclError: no display name and no $DISPLAY environment variable

我很简单想使用savefig（＆＃34; filename.jpg＆＃34;）并且不想使用X服务器来查看情节。

我该如何实现？或者是否有任何其他python库可以生成热图并保存为JPEG文件。

Answer 1

Matplotlib默认使用x-using后端，但您可以通过修改rcParams来设置非交互式后端：

import matplotlib
matplotlib.rcParams['backend'] = "Agg"