DoodlerGAN:可以涂鸦的AI模型

时间:2021-05-07 11:29:18
【文件属性】:
文件名称:DoodlerGAN:可以涂鸦的AI模型
文件大小:1.5MB
文件格式:ZIP
更新时间:2021-05-07 11:29:18
Python 创意草图生成(DoodlerGAN) 论文: : 演示: : 数据集: : 项目页面: : DoodlerGAN是基于零件的生成对抗网络(GAN),旨在生成具有看不见的新颖零件外观组成的创意草图。 具体来说,DoodlerGAN包含两个模块:零件生成器和零件选择器。 给定局部草图的基于零件的表示形式,零件选择器可以预测下一个要绘制的零件类别。 给定零件草图和零件类别的基于零件的表示形式,零件生成器将生成零件的栅格图像(代表零件的外观和位置)。 下面显示了在Doodle GAN上经过Creative Birds和Creative Creatures数据集训练的一些随机选择的世代。 准备 首先,使用Anaconda创建环境。 安装Pytorch以及requirements.txt中列出的其他软件包。 该代码已通过PyTorch 1.3.1和CUDA 10.0进行了测试:
【文件预览】:
DoodlerGAN-main
----data_process.py(12KB)
----generate_creative_creatures.py(25KB)
----requirements.txt(540B)
----CONTRIBUTING.md(1KB)
----LICENSE(1KB)
----inception.py(13KB)
----README.md(5KB)
----generate_creative_birds.py(20KB)
----part_generator.py(38KB)
----part_selector.py(19KB)
----training_scripts()
--------train_creative_creatures()
--------train_creative_birds()
----evaluate.py(14KB)
----CODE_OF_CONDUCT.md(3KB)
----.gitignore(2KB)
----run_part_generator.py(4KB)
----figs()
--------generation.png(1.43MB)
----run_part_selector.py(3KB)

网友评论