TensorFlow中plt.savefig()去除空白区域

【1】关于plt.gcf()

如果你用同一个figure多次显示图片,建议在显示代码的最开始部分加上这句。

plt.gcf()

【2】去除空白区域

****上有不少相关的博客,但是大多数都是只是把空白区域减少了。在这里我就不把采坑的经历写出来,直接展示。

#清除当前figure显示的内容
fig=plt.gcf()

#显示图片
plt.figure(figsize=IMAGE_SIZE)
plt.imshow(image_np)

#去除坐标轴
plt.gca().xaxis.set_major_locator(plt.NullLocator())
plt.gca().yaxis.set_major_locator(plt.NullLocator())

#设置保存路径
out_png_path=os.path.join('mydata/test_result/', 'show{}.png'.format(i))

#保存图片,并设置保存参数
#bbox_inches='tight'和pad_inches=0.0都很关键
#dpi可以调节你保存的图片的清晰度(默认保存的一般清晰度都很感人...)
plt.savefig(out_png_path,bbox_inches='tight',dpi=300,pad_inches=0.0)

【3】参考的博客就懒得写了,感谢各位大佬的指引,这里主要是记录方便日后查看。

最后展示一个结果,如下(建议放大查看):

TensorFlow中plt.savefig()去除空白区域