Python savefig 去掉白色的margin

import matplotlib.pyplot as plt
fig = plt.figure(frameon=False)
fig.set_size_inches(w,h)
ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)
ax.imshow(your_image, aspect='normal')
fig.savefig(fname, dpi)

例如：

fig = plt.figure(frameon=False)
fig.set_size_inches(2.24,2.24)
ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)
ax.imshow(heat_map)
fig.savefig(savepath)

如果需要批量保存多张图片，则可以设置如下：

fig = plt.figure(frameon=False)
fig.set_size_inches(2.24,2.24) # 这样设置输出的为 224 * 224 大小图像
ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)
while True:
	ax.clear() # be careful!
	ax.imshow(heat_map)
	fig.savefig(savepath)

参考文献:
https://stackoverflow.com/questions/8218608/scipy-savefig-without-frames-axes-only-content