
Pix2Pix-Keras:利用Keras实现的动漫图像自动着色(基于pix2pix模型) 2019-2-25
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了使用Keras框架实现的Pix2Pix模型,该模型用于将黑白动漫图片自动上色。通过深度学习技术,可以生成色彩丰富且风格一致的动漫图像。
Pix2Pix-Keras基于pix2pix模型的动漫图片自动上色
环境要求:
- tensorflow-gpu 1.12.0
- Keras 2.2.4
数据集准备:
将训练用的彩色图片放入datasets\OriginalImages文件夹中,然后运行prepare.py进行数据集处理与准备。注意当前datasets中的训练集数量较少,建议增加更多训练样本以防止过拟合。
模型训练:
若要加载预训练权重,请将对应的权重文件放置在weights文件夹内。
在demo.py中新建一个pix2pix模型实例,并调用该实例的train函数进行模型训练。
为新的图片上色:
请先将所需的权重文件放入weights文件夹,然后创建一个新的Pix2Pix模型实例。
全部评论 (0)
还没有任何评论哟~


