United Statesjunyanz/pytorch-CycleGAN-and-pix2pix
Image-to-Image Translation in PyTorch
Thành tựu nổi bật
Những danh hiệu mới nhất của dự án
#1 Repository Của Tuần
#1 Repository Của Tuần
#1 Repository Của Tuần
🖼️ Dịch ảnh thành ảnh: CycleGAN + pix2pix trên PyTorch
Hai mô hình dịch ảnh-to-ảnh tiêu biểu — không cần cặp ảnh đối chiếu — trong một repo PyTorch duy nhất.
Đây là bản cài đặt PyTorch của CycleGAN (dịch ảnh không ghép cặp, ví dụ: ngựa → zebra) và pix2pix (dịch ảnh có ghép cặp, ví dụ: cạnh → ảnh thật). Viết bởi Jun-Yan Zhu và Taesung Park, hỗ trợ DDP cho huấn luyện đa GPU trên một máy.
✨ Điểm nổi bật
- Hỗ trợ Python 3.11 & PyTorch 2.4: Cập nhật năm 2025, tương thích môi trường hiện đại.
- DDP đa GPU: Dùng
torchrun --nproc_per_node=4 train.py ...để huấn luyện song song trên nhiều GPU. - Hai chế độ dịch: CycleGAN (unpaired) và pix2pix (paired), phù hợp nhiều use-case khác nhau.
- Kết quả bằng hoặc tốt hơn bản gốc Torch: So sánh trực tiếp với bản cài đặt Lua/Torch ban đầu.
- Tài liệu & ví dụ đầy đủ: Colab, tips huấn luyện, QA, template mô hình và dataset tùy chỉnh.
- Liên kết nghiên cứu liên quan: CUT (dịch không ghép nhanh), img2img-turbo (một bước, dựa trên StableDiffusion-Turbo).
💡 Vì sao đáng chú ý
Không chỉ là mã nguồn tham khảo, đây là công cụ thực chiến cho nghiên cứu và giảng dạy — từ demo nhanh trên Colab tới huấn luyện đa GPU. Đặc biệt, repo còn chỉ tới các dự án tiếp theo như CUT và img2img-turbo, giúp người dùng dễ dàng tiếp cận công nghệ tiên tiến.
#CycleGAN #pix2pix #PyTorch #ImageTranslation #TiniX
Similar Projects
Discover other similar projects and AI models.
PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]
[CVPR 2021] Anycost GANs for Interactive Image Synthesis and Editing
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.
GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.