文件名称:AttentionedDeepPaint:动画素描图像的自动着色
文件大小:13MB
文件格式:ZIP
更新时间:2024-05-24 19:38:42
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)