plt. Usage of savefig() and save path

Hello everyone , I meet you again , I'm your friend, Quan Jun .

Sometimes images can meet people's visual needs better than data

pytorch There are many ways to save images under , But these are basically based on image processing , Assign a certain dimension to the pixels of the image , See the following blog for details ：

Pytorch How to save pictures in _ Make the future -CSDN Blog _pytorch Save the picture

It is mainly to write some functions to save pictures ;

in addition ,pytorch There are many statements that can directly save pictures

Such as

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

Copy

This statement also needs to convert pixels .

So if

I just need to open a window , Observe the changes of images during training , I have no need for image pixel saving , Just save a window , So the function that I need to save the image is just a

plt.savefig

Copy

plt.savefig And the path to save , And will not be covered during training , Can be on the code for your reference

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

            # 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)

Copy

Be accomplished , You can see the saved pictures

It is already in my save path .

Publisher ： Full stack programmer stack length , Reprint please indicate the source ：https://javaforall.cn/150676.html Link to the original text ：https://javaforall.cn