
pytorch-CycleGAN-and-pix2pix
junyanz
Image-to-Image Translation in PyTorch
AI 简介
这是一个基于PyTorch实现的图像到图像翻译(Image-to-Image Translation)开源工具包,支持CycleGAN(无配对数据)和pix2pix(有配对数据)两类主流生成对抗网络模型。核心功能包括端到端训练/测试、多GPU分布式训练(DDP)、灵活的数据加载与自定义模型扩展;技术特点涵盖稳定收敛的GAN训练策略、模块化代码结构及详尽的使用文档。适用于风格迁移、图像修复、语义图转照片、医学影像增强等需跨域图像映射的研究与工程场景。
Python
Other25.2k
Stars
6.6k
Forks
338
Watchers
566
Issues
Star 增长
今日0
近 7 天0
近 30 天+20
综合评分72
默认分支master