文件名称：inpainting:使用GAN和_或渐进增长训练方案进行部分卷积的图像修复
文件大小：6.17MB
文件格式：ZIP
更新时间：2024-06-02 11:14:51
image-inpainting inpainting progressive-growing partial-convolution Python
图像修复 使用GAN和/或部分卷积以及渐进式增长训练方案进行图像修复。 警告：渐进式增长方案可能会出现问题，并使用更多的gpu内存。 在我的实验中，没有观察到任何改善。 数据集 在实验中，我使用CelebA-HQ数据集。 要创建它，请参阅 。 您可以使用generate_holes.py创建具有多种分辨率的矩形Kong。 对于不规则的Kong， generate_irregular_holes.py是一种创建Kong的方法（可能不是一个好方法）。 建议的配置 progressive_growing = False use_gan = True如果g_input = masked_X ， use_gan = True ； 如果g_input = masked_X+mask则use_gan = False decoder_partial_conv = False pix2pix_st
【文件预览】：
inpainting-master
----.gitignore(14B)
----generate_irregular_holes.py(3KB)
----vgg_16.pth(6.62MB)
----generate_holes.py(2KB)
----README.md(1KB)
----config.py(3KB)
----utest.py(12KB)
----networks.py(27KB)
----main.py(576B)
----model.py(15KB)
----data.py(1KB)