auto color paint for comic using pix2pix and WDSR(keras),based on user:wmylxmj job https://github.com/wmylxmj/Pix2Pix-Keras https://github.com/wmylxmj/Anime-Super-Resolution 感谢大佬的开源项目 漫画、动漫壁纸、本子自动上色,目前先训练了本子的自动上色,时长12个钟,模型效果还行,起码比黑白的好看2333 训练example show: 左边灰度(gray) 中间预测(predict) 右边原图(original)
用训练好后的模型对黑白本子上色效果如下(gray img non-resize predict): 原图original: 预测predict:
how to use:
1.train model 训练自己需要的上色模型:just see wmylxmj job:https://github.com/wmylxmj/Pix2Pix-Keras
2.get colored_img 得到上色后图片:by running code predict.py
项目目录(Directory):
./weights:存放训练后的上色模型以及SR模型,包含wdsr-b-32-x4.h5、discriminator_weights.h5(可省略)、generator_weights.h5
./datasets/OriginalImages:存放用于训练的彩色图片
pix2pix.py:pix2pix model file
WDSR.py:wdsr model file
utils.py: settings of loading data 设置加载数据的方法等
prepare.py: pre-step before you start train 训练前的预处理数据
predict.py: using trained model get colored img 得到上色后图片