plt.savefig()的用法以及保存路径

全栈程序员站长

发布于 2022-06-28 11:51:29

1.9K0

发布于 2022-06-28 11:51:29

文章被收录于专栏：全栈程序员必看

大家好，又见面了，我是你们的朋友全栈君。

图像有时候比数据更能满足人们的视觉需求

pytorch下保存图像有很多种方法，但是这些基本上都是基于图像处理的，将图像的像素指定一定的维度，具体可见以下博客：

Pytorch中保存图片的方式_造未来-CSDN博客_pytorch保存图片

主要是写一些函数来保存图片；

另外，pytorch中有很多可以直接保存图片的语句

如

save_image(fake_images, './img/fake_images-{}.png'.format(epoch + 1))

此语句同样需要转化像素。

那么如果

我只需要打开一个视窗，观察训练过程中图像的变化，我对图像像素保存没有什么需求，只是保存一个视窗，那么我需要的保存图像的函数仅仅是一个

plt.savefig

plt.savefig的用法以及保存的路径，及训练过程中不会被覆盖掉，可以上代码供大家参考

        if epoch % 10== 0:
            plt.title('ber:{:.3f},a: {:.3f},b:{:.3f},snr: {:.3f}'.format(
            error_rate, a, b,M 
            ))
            plt.plot(r3)  # 绘制波形

            # save_image(r3, './img/fake_images-{}.png'.format(epoch + 1))
            # print(type(r3))
            # plt.draw()
            plt.draw()
            plt.savefig('./img/pic-{}.png'.format(epoch + 1))
            plt.pause(1)

            plt.close(fig1)

大功告成，可以看看保存后的图片