上传者: 42139252
|
上传时间: 2022-03-12 22:59:48
|
文件大小: 9.49MB
|
文件类型: -
CycleGAN_ssim
该项目是项目的扩展。
实现和训练有素的周期一致剖成对抗性网络(CycleGAN)如在所描述的具有不同的,具体地SSIM损失,损耗L1,L2损失和它们的组合,以产生更好的视觉质量的图像。
图1:CycleGAN工作
对于CycleGAN实现与L1损失是指。 对于官方CycleGAN执行读取。
先决条件
Python 3.3以上
Tensorflow 1.6+
枕头(PIL)
(可选)
用法
训练模型:
> python train_cycleGAN_loss.py --data_path monet2photo --input_fname_pattern .jpg --model_dir cycleGAN_model --loss_type l1
data_path:具有trainA和trainB文件夹的目录的路径(具有这些特定名称(trainA,tra