- AI源码解读:数字图像处理案例(Python版)
- 李永华编著
- 107字
- 2022-07-27 17:26:17
AI源码解读.数字图像处理案例:Python版
2.3.6 Pix2Pix模型训练及保存
Pix2Pix模型需要分别训练生成器和判别器,按照1:1的次数进行。
1.Pix2Pix模型训练
模型训练相关代码如下:
训练结果如图2-22所示。
图2-22 Pix2Pix训练结果
2.Pix2Pix模型保存
模型保存相关操作如下:
save_path=saver.save(sess,checkpoints_dir+"/model.ckpt",global_step=i)