【文件属性】:
文件名称:AttentionedDeepPaint:动画素描图像的自动着色
文件大小:13MB
文件格式:ZIP
更新时间:2021-05-05 01:52:02
Python
注意DeepPaint
使用生成对抗网络(GAN)自动对动漫草图进行着色
相依性
参考
安装
火炬(> = 0.4.1)(火炬1.0.1中没有问题)
Torchvision(> = 0.2.1)基于您的python版本,操作系统,CUDA版本等...
用法
下载数据集
转到并下载
解压缩
火车
$ ./train.sh
参考
草图图像生成
彩色直方图提取
关注UNet
着色结果示例
要查看实验试用版,请点击
一般结果示例(相同的草图,不同的样式)
注意图结果示例
火车设置和记录
超参数
价值
learning_rate
0.0002
beta1 (optimizer)
0.5
batch_size
4
lambda
100
epochs
30
learning_rate sche
【文件预览】:
AttentionedDeepPaint-master
----models()
--------patch_gan.py(2KB)
--------attention.py(2KB)
--------__init__.py(150B)
--------deepunet.py(6KB)
----colorgram()
--------colorgram.py(6KB)
----colorize.py(3KB)
----train.sh(137B)
----data()
--------test()
--------styles()
----train.py(2KB)
----trainer()
--------__init__.py(45B)
--------trainer.py(970B)
--------deepunet.py(11KB)
----results()
--------README.md(5KB)
----poster.pdf(11.55MB)
----requirements.txt(309B)
----preprocess()
--------sketch.py(1KB)
--------dataloader.py(4KB)
--------__init__.py(301B)
--------image.py(4KB)
----.gitignore(4KB)
----extract_colorgram.py(2KB)
----README.md(3KB)
----utils()
--------io.py(2KB)
--------average.py(762B)
--------losses.py(515B)
--------__init__.py(212B)
--------args.py(3KB)
--------image.py(1KB)