影像有時候比資料更能滿足人們的視覺需求
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)大功告成，可以看看保存后的圖片
已經都整整齊齊的在我的保存路徑下了，
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/336155.html
標籤：AI
上一篇：多智能體強化學習入門
下一篇：深度學習-神經網路卷積核理解