WonwoongCho/GDWCT

🧠 Dịch ảnh theo nhóm với GDWCT — bản cài đặt chính thức từ CVPR 2019

GDWCT (Group-wise Deep Whitening-and-Coloring Transformation) là một phương pháp dịch ảnh sang ảnh dựa trên mạng GAN, cho phép chuyển đổi phong cách ảnh một cách mượt mà và chi tiết.

Đây là mã nguồn chính thức của GDWCT — một mô hình được trình bày tại CVPR 2019 (oral) bởi các nhà nghiên cứu từ Korea University, LG Electronics và Hanyang University. GDWCT sử dụng kỹ thuật Group-wise Deep Whitening-and-Coloring Transformation để tách và truyền lớp phong cách (style) giữa các ảnh, hỗ trợ các bài toán như dịch ảnh mỗi tính năng (e.g., nụ cười ↔ không nụ cười, tóc dài ↔ tóc ngắn).

✨ Điểm nổi bật

• Group-wise Whitening & Coloring: Tách ảnh thành nhiều nhóm đặc trưng, thực hiện biến đổi trống và tô màu theo nhóm để cải thiện chất lượng dịch.
• Hỗ trợ nhiều bộ dữ liệu: Tích hợp sẵn hỗ trợ CelebA, Artworks (monet2photo, cezanne2photo, …) và BAM.
• Mô hình tiền huấn luyện có sẵn: Hỗ trợ tải các mô hình đã huấn luyện trước như Smile ↔ Non-Smile, Bangs ↔ Non-Bangs.
• Cấu hình linh hoạt qua YAML: Tất cả siêu tham số và chế độ chạy đều được cấu hình trong config.yaml.
• So sánh với baseline mạnh: Được kiểm chứng trên CelebA và Artworks, cho kết quả vượt trội hơn các phương pháp truyền thống.
• Chạy đơn giản với một lệnh: Chỉ cần python run.py để bắt đầu huấn luyện hoặc kiểm tra mô hình.

💡 Vì sao đáng chú ý

GDWCT khác biệt so với các phương pháp dịch ảnh thông thường ở chỗ nó không áp dụng biến đổi toàn cục trên toàn bộ không gian đặc trưng, mà chia không gian đó thành các nhóm nhỏ và thực hiện biến đổi độc lập cho từng nhóm. Cách tiếp cận này giúp giảm độ mất thông tin và tăng độ chi tiết của ảnh đầu ra, đặc biệt hiệu quả trong các bài toán dịch ảnh có nhiều thành phần phong cách khác nhau.

Phù hợp cho nghiên cứu trong lĩnh vực dịch ảnh sang ảnh, biến đổi phong cách ảnh, và các ứng dụng thị giác máy tính dựa trên GAN.

---

#GDWCT #PyTorch #ImageTranslation #GAN #ComputerVision #TiniX