junyanz

pytorch-CycleGAN-and-pix2pix

junyanz

Image-to-Image Translation in PyTorch

AI 简介

该项目提供了基于PyTorch的CycleGAN和pix2pix图像到图像转换的实现。核心功能包括支持成对与不成对的图像转换,使用了生成对抗网络(GAN)技术来实现高质量的图像变换。项目最新版本已更新至支持Python 3.11及PyTorch 2.4,并且加入了分布式数据并行(DDP)以提高单机多GPU训练效率。此外,还引入了新的快速图像转换方法img2img-turbo,以及对比学习无配对翻译模型CUT,这些改进使得模型训练更快、更高效。适用于需要进行风格迁移、图像修复、色彩化等计算机视觉任务的场景,尤其是在缺乏大量成对训练数据的情况下。

Python
Other
25.1k
Stars
6.6k
Forks
338
Watchers
566
Issues

Star 增长

今日0
近 7 天+9
近 30 天+48
综合评分79.3
默认分支master