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