Skip to content

XiuSdk/pix2pix-color-paint-for-gray-comic

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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) example1 example2 example3

用训练好后的模型对黑白本子上色效果如下(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 得到上色后图片

About

cartoon/comics/manga colorization 漫画本子卡通上色

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%