Pix2Pix模型需要分别训练生成器和判别器,按照1:1的次数进行。
模型训练相关代码如下:
训练结果如图2-22所示。
图2-22 Pix2Pix训练结果
模型保存相关操作如下:
save_path=saver.save(sess,checkpoints_dir+"/model.ckpt",global_step=i)