AI源码解读:数字图像处理案例(Python版)
上QQ阅读APP看书,第一时间看更新
 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)