动漫素描着色器
使用参考图像自动进行草图着色
先决条件
pytorch
torchvision
numpy
openCV2
matplotlib
数据集
Taebum Kim,“动漫素描着色对”, //www.kaggle.com/ktaebum/anime-sketch-colorization-pair
培养
请参考train.ipynb
测试
请参考test.ipynb
您可以在 )上下载经过培训的检查点。
培训细节
参数
值
学习率
2e-4
批量大小
2
时代
25
优化器
亚当
(beta1,beta2)
(0.5,0.999)
(lambda1,lambd
1