YangNaruto/FQ-GAN
Official implementation of FQ-GAN
🚀 FQ-GAN: Cải tiến GAN bằng Feature Quantization
Triển khai chính thức của FQ-GAN - kỹ thuật Feature Quantization nâng cao hiệu suất huấn luyện GAN.
FQ-GAN là mã nguồn chính thức của bài báo ICML 2020 "Feature Quantization Improves GAN Training". Dự án áp dụng kỹ thuật Feature Quantization vào ba kiến trúc GAN lớn: FQ-BigGAN, FQ-StyleGAN và FQ-U-GAT-IT, giúp cải thiện chất lượng sinh ảnh và ổn định quá trình huấn luyện.
✨ Điểm nổi bật
- FQ-BigGAN: Triển khai dựa trên PyTorchGAN, hỗ trợ huấn luyện trên CIFAR-10/100 và ImageNet với siêu tham số quantization có thể tùy chỉnh.
- FQ-U-GAT-IT: Áp dụng Feature Quantization vào U-GAT-IT cho bài toán dịch ảnh (selfie2anime, cat2dog, horse2zebra...).
- FQ-StyleGAN: Tích hợp quantization vào StyleGAN.
- Mô hình tiền huấn luyện: Cung cấp checkpoint cho Selfie2Anime và Photo2Portrait.
- Colab & Demo: Hỗ trợ huấn luyện và kiểm thử qua Colab và demo trực tuyến.
- Kết quả khảo sát: FQ-BigGAN đạt FID tốt hơn BigGAN truyền thống trên ImageNet.
💡 Vì sao đáng chú ý
FQ-GAN nổi bật bằng cách áp dụng Feature Quantization - một kỹ thuật mã hóa các đặc trưng về dạng rời rạp - vào các GAN tiêu chuẩn, cải thiện đáng kể hiệu suất mà không cần thay đổi kiến trúc lớn. Phương pháp này đặc biệt hiệu quả trong bài toán dịch ảnh và sinh ảnh có độ phân giải cao, nơi sự ổn định của quá trình huấn luyện là yếu tố then chốt.
#FQ-GAN #FeatureQuantization #BigGAN #StyleGAN #U-GAT-IT #ImageTranslation #TiniX
Similar Projects
Discover other similar projects and AI models.
all-MiniLM-L6-v2 This is a sentence-transformers(https://www.SBERT.net) model: It maps sentences & paragraphs to a 384 dimensional dense vector space and can be used for tasks like clustering or semantic search. Usage (Sentence-Transformers) Using t...
Clone a voice in 5 seconds to generate arbitrary speech in real-time
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
CoMoGAN: continuous model-guided image-to-image translation. CVPR 2021 oral.
[ACM MM 2018 Oral] GestureGAN for Hand Gesture-to-Gesture Translation in the Wild
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.