- AI源码解读:数字图像处理案例(Python版)
- 李永华编著
- 107字
- 2022-07-27 17:26:17
AI源码解读.数字图像处理案例:Python版
2.3.6 Pix2Pix模型训练及保存
Pix2Pix模型需要分别训练生成器和判别器,按照1:1的次数进行。
1.Pix2Pix模型训练
模型训练相关代码如下:
![](https://epubservercos.yuewen.com/631FC2/23721638301028906/epubprivate/OEBPS/Images/Figure-P82_24506.jpg?sign=1739421839-RVtCkIuCtDIyDfMQ52IObLyI9nEOPVXZ-0-26dfe8927325029447d8a035f900b7cf)
训练结果如图2-22所示。
![](https://epubservercos.yuewen.com/631FC2/23721638301028906/epubprivate/OEBPS/Images/Figure-P83_6839.jpg?sign=1739421839-NEHSkC7R506n381AXno5fTgnP3pCVI1k-0-453aa0e32e6990d0484b75e87fe16067)
图2-22 Pix2Pix训练结果
2.Pix2Pix模型保存
模型保存相关操作如下:
save_path=saver.save(sess,checkpoints_dir+"/model.ckpt",global_step=i)