TensorFlow笔记（七）——基于matplotlib.pyplot函数的图片显示与保存

╰半橙微兮° 2022-01-07 01:27 653阅读 0赞

注：pyplot函数多用于生成折线图，坐标图等。显示正常图片用PIL库

在上一章《[TensorFlow笔记（六）——图像预处理][TensorFlow]》显示并保存图片时，遇到几个问题：

with tf.Session( ) as sess:    
        image_raw_data =  tf.gfile.FastGFile(pic_path,"rb").read()#pic_path：图片路径
        img_data = tf.image.decode_jpeg(image_raw_data)#解码
        for s in range(6):
                    # 将图像的尺寸调整为224*224
                    result = preprocess_for_train(img_data, 224, 224, None)
                    #print(type(result))
                    #使用pyplot工具可视化得到的图像。
                    plt.imshow(result.eval( ))
                    ll = os.path.join(resize_path,classes[i],b[0]+'_{}'.format(s)+'.jpg')
                    print(ll)
                    plt.savefig(ll)
                    plt.show()#创建一个新的空白图片（坐标轴）所以要保存图片的话应该把这句放在    savefig()函数的后面，否则保存出来的图片是空白的。
                    if s == 6:
                        break

### 问题一： ###

在保存图片时，保存的是一张空白图片，原本的代码：

import matplotlib.pyplot as plt
    
    """ 一些画图代码 """
    
    plt.show()
    plt.savefig("filename.png")

--------------------

### 原因【1】： ###

其实产生这个现象的原因很简单：在 `plt.show()` 后调用了 `plt.savefig()` ，在 `plt.show()` 后实际上已经创建了一个新的空白的图片（坐标轴），这时候你再 `plt.savefig()` 就会保存这个新生成的空白图片。

--------------------

### 解决： ###

1、在 `plt.show()` 之前调用 `plt.savefig()`；

import matplotlib.pyplot as plt
    
    """ 一些画图代码 """
    
    plt.savefig("filename.png")
    plt.show()

2、画图的时候获取当前图像（这一点非常类似于 Matlab 的句柄的概念）：

# gcf: Get Current Figure
        fig = plt.gcf()
        plt.show()
        fig1.savefig('tessstttyyy.png', dpi=100)

### 问题二： ###

解决问题一后，保存的图片有白色的边框，不是我想要的大小，见下图：

![20190625094217191.png][]                      ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3NzY0MTI5_size_16_color_FFFFFF_t_70][]

左边是理想的保存结果（224\*224），右边是实际保存结果（248\*544）

--------------------

### 原理【2】： ###

在用matplotlib画图时，如果图例比较大，画在图中就会挡着线条，这时可以用以下语句把图例画到图外面：

> plt.legend(bbox\_to\_anchor=(1.01, 1), loc=2, borderaxespad=0., handleheight=1.675)

这个语句可以解决图例遮挡线条的问题，同时，也引入了另外的问题：会使savefig保存图片时，不能将图片完整保存，会使图例保存不完整。

我感觉，原因在于：输出的“图”（并不包含图例）是在画布中间的，而输出时，“图”与图例都会一块输出，只有“图”是居中的，所以，图例只能占据画布边缘，所以导致图输出的不完整。

解决办法，用以下语句将“图”往左边缩放一下，给图例腾出空间：

> fig.subplots\_adjust(right=0.7)

另外，设置图片大小可以用以下语句：

> fig.set\_size\_inches(6.4, 4)

--------------------

### 解决： ###

> fig = plt.gcf()  
> plt.subplots\_adjust(top=1,bottom=0,left=0,right=1,hspace=0,wspace=0)\#以左下角为顶点，top/right的值取（0，1\]之间，相当于生成图片占画框的百分比  
> fig.set\_size\_inches(2.24, 2.24)\#设置输出图片大小，长、宽=实际大小/100  
> fig.savefig(ll, dpi=100) \#保存图片  
> print(ll)

--------------------

### 参考： ###

【1】解决使用 plt.savefig 保存图片时一片空白：[https://blog.csdn.net/u010099080/article/details/52912439][https_blog.csdn.net_u010099080_article_details_52912439]

【2】matplotlib savefig 保存图片大小：https://blog.csdn.net/o1101574955/article/details/70543915

【3】[Python 数据分析与展示(北京理工大学 )][Python _]：该教学视频适合入门学习

【4】[Python之Matplotlib库常用函数大全（含注释）][Python_Matplotlib]：根据该视频做的总结

[TensorFlow]: https://mp.blog.csdn.net/postedit/93361975
[20190625094217191.png]: https://img-blog.csdnimg.cn/20190625094217191.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3NzY0MTI5_size_16_color_FFFFFF_t_70]: /images/20211227/52ffdcae28904e349d3356086ce3de1a.png
[https_blog.csdn.net_u010099080_article_details_52912439]: https://blog.csdn.net/u010099080/article/details/52912439
[Python _]: https://www.bilibili.com/video/av10101509/
[Python_Matplotlib]: https://blog.csdn.net/u011995719/article/details/71124227