lightweight-gan

收录理由

如果你的图片集不大，GPU资源也有限，这个基于PyTorch实现的ICLR 2021轻量级GAN，给出了一个务实的高分辨率图像合成方案。它的核心设计——生成器里的跳跃层激励配合判别器中的自监督学习——让模型在单块GPU上训练数小时就能收敛，即使数据集只有不到一百张图也能出效果。使用时只需一条命令，指定数据文件夹和输出尺寸，模型会自动保存检查点和生成样本。内置的数据增强功能对低数据训练很关键，也能在命令行里直接调整。代码是研究性质，有些边角可能不够打磨，但胜在直接可用，想亲手跑GAN又不想烧太多算力的话，这是个实在的选择。