上传者: 42120550
|
上传时间: 2021-11-23 20:01:23
|
文件大小: 326.77MB
|
文件类型: -
Pix2Pix-Keras
基于pix2pix模型的动漫图片自动上色 2019-2-25
环境
tensorflow-gpu 1.12.0
Keras 2.2.4
数据集的准备:
把训练的彩色图片放入datasets\OriginalImages文件夹
运行prepare.py进行数据集的处理与准备
注意:当前datasets中的训练集过少,可增加训练集进行训练,防止过拟合
训练模型:
若要加载预训练权重,将权重文件放入weights文件夹
最新权重(在30020张图片的大数据集下训练后的权重)下载
链接: 提取码:cac2
在demo.py中新建一个pix2pix模型实例
调用实例中的train函数进行训练
为新的图片上色:
将权重文件放入weights文件夹
新建一个pix2pix模型实例