Anime Sketch Colorizer:带有参考图像的自动草图着色 源码

动漫素描着色器 使用参考图像自动进行草图着色 先决条件 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